| 136 | } |
| 137 | |
| 138 | static void BM_Adagrad(int iters, int params) { |
| 139 | const int64 tot = static_cast<int64>(iters) * params; |
| 140 | testing::ItemsProcessed(tot); |
| 141 | testing::BytesProcessed(tot * sizeof(float)); |
| 142 | Graph* init; |
| 143 | Graph* train; |
| 144 | Adagrad(params, &init, &train); |
| 145 | test::Benchmark("cpu", train, GetOptions(), init).Run(iters); |
| 146 | } |
| 147 | BENCHMARK(BM_Adagrad)->Arg(128 << 10)->Arg(256 << 10); |
| 148 | |
| 149 | static void SparseAdagrad(int32 m, int32 n, Graph** init_g, Graph** train_g) { |
nothing calls this directly
no test coverage detected