| 23 | |
| 24 | template <typename T> |
| 25 | static Graph* LargeOneDCumsum(int num_x, bool reverse = false) { |
| 26 | auto* g = new Graph(OpRegistry::Global()); |
| 27 | Tensor data(DataTypeToEnum<T>::value, TensorShape({num_x})); |
| 28 | data.flat<T>().setRandom(); |
| 29 | Tensor axes(DT_INT32, TensorShape({})); |
| 30 | axes.flat<int32>()(0) = 0; |
| 31 | test::graph::Cumsum(g, test::graph::Constant(g, data), |
| 32 | test::graph::Constant(g, axes)); |
| 33 | return g; |
| 34 | } |
| 35 | |
| 36 | static Graph* ColCumsum(int num_x, int num_y, bool reverse = false) { |
| 37 | auto* g = new Graph(OpRegistry::Global()); |
nothing calls this directly
no test coverage detected