| 298 | } |
| 299 | |
| 300 | static void BM_RMSProp(int iters, int params) { |
| 301 | const int64 tot = static_cast<int64>(iters) * params; |
| 302 | testing::ItemsProcessed(tot); |
| 303 | testing::BytesProcessed(tot * sizeof(float)); |
| 304 | Graph* init; |
| 305 | Graph* train; |
| 306 | RMSProp(params, &init, &train); |
| 307 | test::Benchmark("cpu", train, GetOptions(), init).Run(iters); |
| 308 | } |
| 309 | BENCHMARK(BM_RMSProp)->Arg(128 << 10)->Arg(256 << 10); |
| 310 | |
| 311 | static void AddSign(int32 n, Graph** init_g, Graph** train_g) { |
nothing calls this directly
no test coverage detected