signatures = root.get_c.get_concrete_function()
  save.save(root, non_chunked_dir, signatures=signatures,
            options=save_options.SaveOptions(experimental_image_format=False))


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")

        "@absl_py//absl:app",
    ],
)

# copybara:uncomment_begin(google-only)
#
# py_strict_binary(
#     name = "testdata/generate_chunked_models",
#     srcs = ["testdata/generate_chunked_models.py"],
#     python_version = "PY3",
#     srcs_version = "PY3",
#     deps = [
#         "//tensorflow/python/compat:v2_compat",
#         "//tensorflow/python/eager:def_function",