| 399 | } // namespace tflite |
| 400 | |
| 401 | int main(int argc, char** argv) { |
| 402 | tensorflow::string model_file; |
| 403 | const std::vector<tensorflow::Flag> flag_list = { |
| 404 | tensorflow::Flag("test_model_file", &model_file, |
| 405 | "Path to test tflite model file."), |
| 406 | }; |
| 407 | |
| 408 | const bool parse_result = tensorflow::Flags::Parse(&argc, argv, flag_list); |
| 409 | if (!parse_result) { |
| 410 | std::cerr << "Required test_model_file\n"; |
| 411 | std::abort(); |
| 412 | } |
| 413 | g_test_model_dir = |
| 414 | new tensorflow::string(tensorflow::io::Dirname(model_file)); |
| 415 | ::tensorflow::port::InitMain(argv[0], &argc, &argv); |
| 416 | return RUN_ALL_TESTS(); |
| 417 | } |