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

Function BM_cpu_RandomGamma

tensorflow/core/kernels/random_op_test.cc:87–93  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

85}
86
87void BM_cpu_RandomGamma(int iters, int nsamp, int nalpha) {
88 testing::ItemsProcessed(static_cast<int64>(iters) * nsamp * nalpha);
89 Graph* g = new Graph(OpRegistry::Global());
90 test::graph::RandomGamma(g, test::graph::Constant(g, VecShape(nsamp)),
91 test::graph::Constant(g, VecAlphas(nalpha)));
92 test::Benchmark("cpu", g).Run(iters);
93}
94BENCHMARK(BM_cpu_RandomGamma)->RangePair(1 << 14, 4 << 15, 2, 50);
95
96void BM_PhiloxRandom(int iters) {

Callers

nothing calls this directly

Calls 7

ItemsProcessedFunction · 0.85
VecAlphasFunction · 0.85
ConstantFunction · 0.70
VecShapeFunction · 0.70
RandomGammaFunction · 0.50
BenchmarkClass · 0.50
RunMethod · 0.45

Tested by

no test coverage detected