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

Function DoRowCumsum

tensorflow/core/kernels/scan_ops_test.cc:77–83  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

75}
76
77static void DoRowCumsum(int iters, const string& device, int num_x, int num_y,
78 bool reverse = false) {
79 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y);
80 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y *
81 sizeof(float));
82 test::Benchmark(device, RowCumsum(num_x, num_y, reverse)).Run(iters);
83}
84
85static void DoColCumsum(int iters, const string& device, int num_x, int num_y,
86 bool reverse = false) {

Callers 2

BM_Sum2DRowCumsumGPUFunction · 0.85

Calls 5

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

Tested by

no test coverage detected