Function run_main

tensorflow/lite/python/tflite_convert.py:492–511

Main in toco_convert.py.

(_)

def run_main(_):
  """Main in toco_convert.py."""
  if tf2.enabled():
    parser = _get_tf2_parser()
  else:
    parser = _get_tf1_parser()

  tflite_flags, unparsed = parser.parse_known_args(args=sys.argv[1:])

  if tf2.enabled():
    _convert_tf2_model(tflite_flags)
  else:
    try:
      _check_tf1_flags(tflite_flags, unparsed)
    except ValueError as e:
      parser.print_usage()
      file_name = os.path.basename(sys.argv[0])
      sys.stderr.write("{0}: error: {1}\n".format(file_name, str(e)))
      sys.exit(1)
    _convert_tf1_model(tflite_flags)


def main():