| 104 | } |
| 105 | |
| 106 | static void BM_SGD(int iters, int params) { |
| 107 | const int64 tot = static_cast<int64>(iters) * params; |
| 108 | testing::ItemsProcessed(tot); |
| 109 | testing::BytesProcessed(tot * sizeof(float)); |
| 110 | Graph* init; |
| 111 | Graph* train; |
| 112 | SGD(params, &init, &train); |
| 113 | test::Benchmark("cpu", train, GetOptions(), init).Run(iters); |
| 114 | } |
| 115 | BENCHMARK(BM_SGD)->Arg(128 << 10)->Arg(256 << 10); |
| 116 | |
| 117 | static void Adagrad(int32 n, Graph** init_g, Graph** train_g) { |
nothing calls this directly
no test coverage detected