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

Function BM_Sum2DRowReduceGPU

tensorflow/core/kernels/reduction_ops_test.cc:142–144  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

140BENCHMARK(BM_Sum2DToScalarGPUHalf)->RangePair(1, 8192, 1, 8192);
141
142static void BM_Sum2DRowReduceGPU(int iters, int num_x, int num_y) {
143 DoRowReduce(iters, "gpu", "Sum", num_x, num_y);
144}
145BENCHMARK(BM_Sum2DRowReduceGPU)->RangePair(1, 8192, 1, 8192);
146
147static void BM_Sum2DColumnReduceGPU(int iters, int num_x, int num_y) {

Callers

nothing calls this directly

Calls 1

DoRowReduceFunction · 0.85

Tested by

no test coverage detected