def generate_chunked_model(non_chunked_dir: str, chunked_dir: str):
  saved_model = loader_impl.parse_saved_model(non_chunked_dir)
  prefix = file_io.join(compat.as_str(chunked_dir), "saved_model")
  file_io.write_string_to_file(f"{prefix}.pbtxt", str(saved_model))
  proto_splitter.SavedModelSplitter(saved_model).write(prefix)


def main(argv: Sequence[str]) -> None:
  if len(argv) > 1:

    asset_dir = self.create_tempdir('assets').full_path
    asset_file = os.path.join(asset_dir, 'vocab_file.txt')
    content = '\n'.join(['static', 'range', 'quantization'])
    file_io.write_string_to_file(filename=asset_file, file_content=content)

    # The resulting table looks like:
    # "static" -> 0
    # "range" -> 1
    # "quantization" -> 2
    # default -> -1

    )
    asset_filename = 'assets.extra/tf_serving_warmup_requests'
    file_io.create_dir_v2(
        os.path.join(self._input_saved_model_path, 'assets.extra')
    )
    file_io.write_string_to_file(
        filename=os.path.join(self._input_saved_model_path, asset_filename),
        file_content='Test content',
    )

    tags = {tag_constants.SERVING}