| 50 | } |
| 51 | |
| 52 | Node* Var(Graph* const g, const int n) { |
| 53 | return test::graph::Var(g, DT_FLOAT, TensorShape({n})); |
| 54 | } |
| 55 | |
| 56 | // Returns a vector of size 'nodes' with each node being of size 'node_size'. |
| 57 | std::vector<Node*> VarVector(Graph* const g, const int nodes, |