diff options
Diffstat (limited to 'tensorflow/core/kernels/sdca_ops_test.cc')
| -rw-r--r-- | tensorflow/core/kernels/sdca_ops_test.cc | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/sdca_ops_test.cc b/tensorflow/core/kernels/sdca_ops_test.cc index 9ddbd817e1..400f330ce7 100644 --- a/tensorflow/core/kernels/sdca_ops_test.cc +++ b/tensorflow/core/kernels/sdca_ops_test.cc @@ -232,6 +232,17 @@ void BM_SDCA(const int iters, const int num_examples) { test::Benchmark("cpu", train, GetSingleThreadedOptions(), init).Run(iters); } +void BM_SDCA_LARGE_DENSE(const int iters, const int num_examples) { + testing::StopTiming(); + Graph* init = nullptr; + Graph* train = nullptr; + GetGraphs(num_examples, 0 /* sparse feature groups */, + 0 /* sparse features per group */, 5 /* dense feature groups*/, + 200000 /* dense features per group */, &init, &train); + testing::StartTiming(); + test::Benchmark("cpu", train, GetSingleThreadedOptions(), init).Run(iters); +} + void BM_SDCA_LARGE_SPARSE(const int iters, const int num_examples) { testing::StopTiming(); Graph* init = nullptr; @@ -242,10 +253,10 @@ void BM_SDCA_LARGE_SPARSE(const int iters, const int num_examples) { testing::StartTiming(); test::Benchmark("cpu", train, GetMultiThreadedOptions(), init).Run(iters); } - } // namespace BENCHMARK(BM_SDCA)->Arg(128)->Arg(256)->Arg(512)->Arg(1024); +BENCHMARK(BM_SDCA_LARGE_DENSE)->Arg(128)->Arg(256)->Arg(512)->Arg(1024); BENCHMARK(BM_SDCA_LARGE_SPARSE)->Arg(128)->Arg(256)->Arg(512)->Arg(1024); } // namespace tensorflow |