path = os.path.join(TESTDATA_PATH.value, MODEL_NAME.value)
  saved_model.save(model, path)


if __name__ == "__main__":
  v2_compat.enable_v2_behavior()
  flags.mark_flag_as_required("path")
  flags.mark_flag_as_required("model_name")

    'output', None,
    'Path to write the genereated register op file and MLIR file.')

flags.DEFINE_bool('gen_register_op', True,
                  'Generate register op cc file or tfr mlir file.')

flags.mark_flag_as_required('output')


@Composite('TestRandom', derived_attrs=['T: numbertype'], outputs=['o: T'])
def _composite_random_op():
  pass


def main(_):
  if FLAGS.gen_register_op:

_SRCS = flags.DEFINE_string('src', None, 'source file locations')
_NAMESPACE = flags.DEFINE_string('namespace', 'mlir::quant',
                                 'namespace in the generated file')

flags.mark_flags_as_required(['output_file', 'src'])


def _substitute_for_loop_template(module: str) -> str:
  """Substitutes the for loop templates in the given module."""
  compiled_regex = re.compile(