MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / BM_Sum3DYReduceGPU

Function BM_Sum3DYReduceGPU

tensorflow/core/kernels/reduction_ops_test.cc:152–154  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

150BENCHMARK(BM_Sum2DColumnReduceGPU)->RangePair(1, 8192, 1, 8192);
151
152static void BM_Sum3DYReduceGPU(int iters, int num_x, int num_y) {
153 Do3DYReduce(iters, "gpu", "Sum", num_x, num_y);
154}
155BENCHMARK(BM_Sum3DYReduceGPU)->RangePair(64, 4096, 64, 4096);
156
157static void BM_Sum3DXZReduceGPU(int iters, int num_x, int num_y) {

Callers

nothing calls this directly

Calls 1

Do3DYReduceFunction · 0.85

Tested by

no test coverage detected