Returns:
    Signature def map where the values (`SignatureDef`) are serialized.
  """
  signature_def_map_serialized = {}
  for key, signature_def in signature_def_map.items():
    signature_def_map_serialized[key] = signature_def.SerializeToString()

  return signature_def_map_serialized


def _has_quantization_method(
    quantization_specs: qc.QuantizationSpecs, method: str
) -> bool:

# LINT.IfChange(quantize_qat_model)
def quantize_qat_model(
    src_saved_model_path: str,
    dst_saved_model_path: str,
    quantization_options_serialized: bytes,
    *,
    signature_keys: list[str],
    signature_def_map_serialized: dict[str, bytes],
    py_function_library: py_function_lib.PyFunctionLibrary,
) -> Any: ...  # Status

# LINT.ThenChange()

# LINT.IfChange(quantize_ptq_dynamic_range)
def quantize_ptq_dynamic_range(

      serialized `SignatureDef` mapping for the `signature_def_map_serialized`
      argument.
      )pbdoc",
      py::arg("src_saved_model_path"), py::arg("dst_saved_model_path"),
      py::arg("quantization_options_serialized"), py::kw_only(),
      py::arg("signature_keys"), py::arg("signature_def_map_serialized"),
      py::arg("py_function_library"));

  m.def(

        serialized `SignatureDef` mapping for the `signature_def_map_serialized`
        argument.

        Raises `StatusNotOk` exception if when the run was unsuccessful.
        )pbdoc",
        py::arg("src_saved_model_path"), py::arg("dst_saved_model_path"),
        py::arg("quantization_config_serialized"), py::kw_only(),
        py::arg("signature_keys"), py::arg("signature_def_map_serialized"),
        py::arg("py_function_library"));

# LINT.IfChange(static_range_ptq)
def static_range_ptq(
    src_saved_model_path: str,
    dst_saved_model_path: str,
    quantization_config_serialized: bytes,
    *,
    signature_keys: list[str],
    signature_def_map_serialized: dict[str, bytes],
    py_function_library: py_function_lib.PyFunctionLibrary,
) -> Any: ...  # Status

# LINT.ThenChange()

# LINT.IfChange(weight_only_ptq)
def weight_only_ptq(

      pywrap_quantize_model.quantize_ptq_static_range(
          src_saved_model_path,
          dst_saved_model_path,
          quant_opts_serialized,
          signature_keys=signature_def_keys,
          signature_def_map_serialized={},
          function_aliases={},
          py_function_library=py_function_lib.PyFunctionLibrary(),
          representative_dataset_file_map_serialized=None,
      )

  Returns:
    Signature def map where the values (`SignatureDef`) are serialized.
  """
  signature_def_map_serialized = {}
  for key, signature_def in signature_def_map.items():
    signature_def_map_serialized[key] = signature_def.SerializeToString()

  return signature_def_map_serialized


def _save_representative_dataset(
    representative_dataset: repr_dataset.RepresentativeDatasetOrMapping,