| 169 | } |
| 170 | } |
| 171 | static void BM_SparseAdagrad(int iters, int m, int n) { |
| 172 | const int64 tot = static_cast<int64>(iters) * m * n; |
| 173 | testing::UseRealTime(); |
| 174 | testing::ItemsProcessed(tot); |
| 175 | testing::BytesProcessed(tot * sizeof(float)); |
| 176 | Graph* init; |
| 177 | Graph* train; |
| 178 | SparseAdagrad(m, n, &init, &train); |
| 179 | test::Benchmark("cpu", train, GetMultiThreadedOptions(), init).Run(iters); |
| 180 | } |
| 181 | BENCHMARK(BM_SparseAdagrad) |
| 182 | ->ArgPair(128, 1 << 10) |
| 183 | ->ArgPair(128, 4 << 10) |
nothing calls this directly
no test coverage detected