| 372 | } |
| 373 | |
| 374 | static void BM_PowerSign(int iters, int params) { |
| 375 | const int64 tot = static_cast<int64>(iters) * params; |
| 376 | testing::ItemsProcessed(tot); |
| 377 | testing::BytesProcessed(tot * sizeof(float)); |
| 378 | Graph* init; |
| 379 | Graph* train; |
| 380 | PowerSign(params, &init, &train); |
| 381 | test::Benchmark("cpu", train, GetOptions(), init).Run(iters); |
| 382 | } |
| 383 | BENCHMARK(BM_PowerSign)->Arg(128 << 10)->Arg(256 << 10); |
| 384 | |
| 385 | } // end namespace tensorflow |
nothing calls this directly
no test coverage detected