| 50 | } |
| 51 | |
| 52 | static Node* Zeros(Graph* g, int n) { |
| 53 | Tensor data(DT_FLOAT, TensorShape({n})); |
| 54 | data.flat<float>().setZero(); |
| 55 | return test::graph::Constant(g, data); |
| 56 | } |
| 57 | |
| 58 | static Node* Zeros(Graph* g, int m, int n) { |
| 59 | Tensor data(DT_FLOAT, TensorShape({m, n})); |