| 979 | } // namespace tflite |
| 980 | |
| 981 | int main(int argc, char** argv) { |
| 982 | tensorflow::string model_file; |
| 983 | const std::vector<tensorflow::Flag> flag_list = { |
| 984 | tensorflow::Flag("test_model_file", &model_file, |
| 985 | "Path to test tflite model file."), |
| 986 | }; |
| 987 | |
| 988 | const bool parse_result = tensorflow::Flags::Parse(&argc, argv, flag_list); |
| 989 | if (!parse_result) { |
| 990 | std::cerr << "Required test_model_file\n"; |
| 991 | std::abort(); |
| 992 | } |
| 993 | g_test_model_dir = |
| 994 | new tensorflow::string(tensorflow::io::Dirname(model_file)); |
| 995 | ::tensorflow::port::InitMain(argv[0], &argc, &argv); |
| 996 | return RUN_ALL_TESTS(); |
| 997 | } |