Marat Dukhan | c08cdf5 | 2019-12-09 09:17:51 -0800 | [diff] [blame] | 1 | // Copyright 2019 Google LLC |
| 2 | // |
| 3 | // This source code is licensed under the BSD-style license found in the |
| 4 | // LICENSE file in the root directory of this source tree. |
| 5 | |
| 6 | #pragma once |
| 7 | |
| 8 | #include "models/models.h" |
| 9 | #include <benchmark/benchmark.h> |
| 10 | |
| 11 | |
| 12 | #define BENCHMARK_END2END(benchmark_fn) \ |
| 13 | BENCHMARK_CAPTURE(benchmark_fn, mobilenet_v1, models::MobileNetV1)->Unit(benchmark::kMicrosecond)->UseRealTime(); \ |
| 14 | BENCHMARK_CAPTURE(benchmark_fn, mobilenet_v2, models::MobileNetV2)->Unit(benchmark::kMicrosecond)->UseRealTime(); \ |
| 15 | BENCHMARK_CAPTURE(benchmark_fn, mobilenet_v3_large, models::MobileNetV3Large)->Unit(benchmark::kMicrosecond)->UseRealTime(); \ |
Marat Dukhan | 77b78a6 | 2019-12-10 23:32:49 -0800 | [diff] [blame] | 16 | BENCHMARK_CAPTURE(benchmark_fn, mobilenet_v3_small, models::MobileNetV3Small)->Unit(benchmark::kMicrosecond)->UseRealTime(); |