| 56 | } |
| 57 | |
| 58 | static Graph* ThreeDYCumsum(int num_y, int num_z, bool reverse = false) { |
| 59 | auto* g = new Graph(OpRegistry::Global()); |
| 60 | Tensor data(DT_FLOAT, TensorShape({32, num_y, num_z})); |
| 61 | data.flat<float>().setRandom(); |
| 62 | Tensor axes(DT_INT32, TensorShape({})); |
| 63 | axes.flat<int32>()(0) = 1; |
| 64 | test::graph::Cumsum(g, test::graph::Constant(g, data), |
| 65 | test::graph::Constant(g, axes)); |
| 66 | return g; |
| 67 | } |
| 68 | |
| 69 | template <typename T> |
| 70 | static void LargeOneDimensional(int iters, const string& device, int num_x, |
no test coverage detected