| 21 | namespace benchmark { |
| 22 | |
| 23 | int Main(int argc, char** argv) { |
| 24 | ::tflite::InitTensorFlow(); |
| 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 | benchmark.Run(argc, argv); |
| 34 | return 0; |
| 35 | } |
| 36 | } // namespace benchmark |
| 37 | } // namespace tflite |
| 38 |
no test coverage detected