Function Do3DYCumsum

tensorflow/core/kernels/scan_ops_test.cc:93–99

}

static void Do3DYCumsum(int iters, const string& device, int num_x, int num_y,
                        bool reverse = false) {
  testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y);
  testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y *
                          sizeof(float));
  test::Benchmark(device, ThreeDYCumsum(num_x, num_y, reverse)).Run(iters);
}

static void BM_OneDCumsumGPU(int iters, int num_x) {
  LargeOneDimensional<float>(iters, "gpu", num_x);