| 209 | } |
| 210 | |
| 211 | static void BM_Momentum(int iters, int params) { |
| 212 | const int64 tot = static_cast<int64>(iters) * params; |
| 213 | testing::ItemsProcessed(tot); |
| 214 | testing::BytesProcessed(tot * sizeof(float)); |
| 215 | Graph* init; |
| 216 | Graph* train; |
| 217 | Momentum(params, &init, &train); |
| 218 | test::Benchmark("cpu", train, GetOptions(), init).Run(iters); |
| 219 | } |
| 220 | BENCHMARK(BM_Momentum)->Arg(128 << 10)->Arg(256 << 10); |
| 221 | |
| 222 | static void Adam(int32 n, Graph** init_g, Graph** train_g) { |
nothing calls this directly
no test coverage detected