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

Function BM_Sum3DXZReduceGPU

tensorflow/core/kernels/reduction_ops_test.cc:157–159  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

155BENCHMARK(BM_Sum3DYReduceGPU)->RangePair(64, 4096, 64, 4096);
156
157static void BM_Sum3DXZReduceGPU(int iters, int num_x, int num_y) {
158 Do3DXZReduce(iters, "gpu", "Sum", num_x, num_y);
159}
160BENCHMARK(BM_Sum3DXZReduceGPU)->RangePair(64, 4096, 64, 4096);
161
162static void BM_Mean2DToScalarGPU(int iters, int num_x, int num_y) {

Callers

nothing calls this directly

Calls 1

Do3DXZReduceFunction · 0.85

Tested by

no test coverage detected