Function main

tensorflow/lite/tools/gen_op_registration_main.cc:98–117

}  // namespace

int main(int argc, char** argv) {
  string input_model;
  string output_registration;
  string tflite_path;
  ParseFlagAndInit(argc, argv, &input_model, &output_registration,
                   &tflite_path);

  std::vector<string> builtin_ops;
  std::vector<string> custom_ops;
  std::ifstream fin(input_model);
  std::stringstream content;
  content << fin.rdbuf();
  // Need to store content data first, otherwise, it won't work in bazel.
  string content_str = content.str();
  const ::tflite::Model* model = ::tflite::GetModel(content_str.data());
  ::tflite::ReadOpsFromModel(model, &builtin_ops, &custom_ops);
  GenerateFileContent(tflite_path, output_registration, builtin_ops,
                      custom_ops);
  return 0;
}