| 16 | #include "tensorflow/lite/testing/kernel_test/util.h" |
| 17 | |
| 18 | int main(int argc, char** argv) { |
| 19 | tflite::testing::kernel_test::TestOptions options = |
| 20 | tflite::testing::kernel_test::ParseTfliteKernelTestFlags(&argc, argv); |
| 21 | const bool run_reference_kernel = options.kernel_type == "REFERENCE"; |
| 22 | const tflite::testing::TfLiteDriver::DelegateType delegate_type = |
| 23 | options.kernel_type == "NNAPI" |
| 24 | ? tflite::testing::TfLiteDriver::DelegateType::kNnapi |
| 25 | : tflite::testing::TfLiteDriver::DelegateType::kNone; |
| 26 | |
| 27 | auto runner = absl::make_unique<tflite::testing::TfLiteDriver>( |
| 28 | delegate_type, run_reference_kernel); |
| 29 | if (tflite::testing::kernel_test::RunKernelTest(options, runner.get()) == |
| 30 | kTfLiteOk) { |
| 31 | return 0; |
| 32 | } |
| 33 | |
| 34 | return -1; |
| 35 | } |
nothing calls this directly
no test coverage detected