}
        py_function_library.SaveExportedModel(
            dst_saved_model_path, *exported_model, src_saved_model_path, tags,
            signature_def_map);

        return absl::OkStatus();
      },
      R"pbdoc(
      Quantizes a model that went through quantization-aware training (QAT)
      saved at `src_saved_model_path`. The resulting model will be saved to

def _run_static_range_qat(
    src_saved_model_path: str,
    dst_saved_model_path: str,
    quant_opts: _QuantizationOptions,
    signature_def_map: _SignatureDefMap,
) -> None:
  """Runs static-range quantization for a Quantization-Aware Trained model.

  Runs the quantization for a model trained using QAT.

  Args:
    src_saved_model_path: Path to the source SavedModel directory.

    exported_model: exported_model_pb2.ExportedModel,
    src_saved_model_path: str,
    dst_saved_model_path: str,
    signature_def_map: Mapping[str, meta_graph_pb2.SignatureDef],
    tags: Collection[str],
) -> bool:
  """Saves the model and copies the assets from the source model.

  Args:
    exported_model: ExportedModel to save.
    src_saved_model_path: Path to the source SavedModel. This will be used to