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

Function DoRowReduce

tensorflow/core/kernels/reduction_ops_test.cc:95–101  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

93}
94
95static void DoRowReduce(int iters, const string& device, const string& reduce,
96 int num_x, int num_y) {
97 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y);
98 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y *
99 sizeof(float));
100 test::Benchmark(device, RowReduce(reduce, num_x, num_y)).Run(iters);
101}
102
103static void DoColReduce(int iters, const string& device, const string& reduce,
104 int num_x, int num_y) {

Callers 1

BM_Sum2DRowReduceGPUFunction · 0.85

Calls 5

ItemsProcessedFunction · 0.85
BytesProcessedFunction · 0.85
RowReduceFunction · 0.85
BenchmarkClass · 0.50
RunMethod · 0.45

Tested by

no test coverage detected