| 536 | } // namespace tflite |
| 537 | |
| 538 | int main(int argc, char** argv) { |
| 539 | tensorflow::string model_file; |
| 540 | const std::vector<tensorflow::Flag> flag_list = { |
| 541 | tensorflow::Flag("test_model_file", &model_file, |
| 542 | "Path to test tflite model file."), |
| 543 | }; |
| 544 | |
| 545 | const bool parse_result = tensorflow::Flags::Parse(&argc, argv, flag_list); |
| 546 | if (!parse_result) { |
| 547 | std::cerr << "Required test_model_file\n"; |
| 548 | std::abort(); |
| 549 | } |
| 550 | g_test_model_dir = |
| 551 | new tensorflow::string(tensorflow::io::Dirname(model_file)); |
| 552 | ::tensorflow::port::InitMain(argv[0], &argc, &argv); |
| 553 | return RUN_ALL_TESTS(); |
| 554 | } |