| 22 | namespace benchmark { |
| 23 | |
| 24 | int Main(int argc, char** argv) { |
| 25 | #ifdef TFLITE_CUSTOM_OPS_HEADER |
| 26 | TFLITE_LOG(INFO) << "STARTING with custom ops!"; |
| 27 | #else |
| 28 | TFLITE_LOG(INFO) << "STARTING!"; |
| 29 | #endif |
| 30 | BenchmarkTfLiteModel benchmark; |
| 31 | BenchmarkLoggingListener listener; |
| 32 | benchmark.AddListener(&listener); |
| 33 | if (benchmark.Run(argc, argv) != kTfLiteOk) { |
| 34 | TFLITE_LOG(ERROR) << "Benchmarking failed."; |
| 35 | return EXIT_FAILURE; |
| 36 | } |
| 37 | return EXIT_SUCCESS; |
| 38 | } |
| 39 | } // namespace benchmark |
| 40 | } // namespace tflite |
| 41 |
no test coverage detected