Function BM_MulChain

tensorflow/core/kernels/basic_ops_benchmark_test.cc:67–74

Benchmark a chain of simple multiplications. This emphasizes per-op overhead.

// Benchmark a chain of simple multiplications.
// This emphasizes per-op overhead.
static void BM_MulChain(int iters, int chain_length) {
  const int64 tot = static_cast<int64>(iters) * chain_length;
  testing::ItemsProcessed(tot);
  Graph* init;
  Graph* run;
  MulChain(chain_length, &init, &run);
  test::Benchmark("cpu", run, GetOptions(), init).Run(iters);
}
BENCHMARK(BM_MulChain)->Arg(1 << 10);

}  // end namespace tensorflow