code: 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.
        dst_saved_model_path: Path to the destination SavedModel directory.
        quant_opts: Quantization options.
        signature_def_map: Signature def key -> SignatureDef mapping.
      """
      logging.info("Running static-range quantization for QAT model.")
  
      loader = saved_model_loader.SavedModelLoader(src_saved_model_path)
      function_aliases = loader.get_meta_graph_def_from_tags(
          quant_opts.tags
      ).meta_info_def.function_aliases
  
      pywrap_quantize_model.quantize_qat_model(
          src_saved_model_path,
          dst_saved_model_path,
          quantization_options_serialized=quant_opts.SerializeToString(),
          signature_keys=list(quant_opts.signature_keys),
          signature_def_map_serialized=_serialize_signature_def_map(
              signature_def_map
          ),
          function_aliases=dict(function_aliases),
          py_function_library=py_function_lib.PyFunctionLibrary(),
      )
  
imports: import tempfile
  from typing import Mapping, Optional
  from absl import logging
  from tensorflow.compiler.mlir.quantization.tensorflow import quantization_options_pb2
  from tensorflow.compiler.mlir.quantization.tensorflow.python import py_function_lib
  from tensorflow.compiler.mlir.quantization.tensorflow.python import pywrap_quantize_model
  from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset
  from tensorflow.compiler.mlir.quantization.tensorflow.python import save_model
  from tensorflow.core.protobuf import meta_graph_pb2
  from tensorflow.python.lib.io import file_io
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import loader_impl
  from tensorflow.python.saved_model import signature_constants
  from tensorflow.python.saved_model import tag_constants
  from tensorflow.python.trackable import autotrackable
  from tensorflow.python.util import tf_export
label_desc: Use image node_id 3 for calling a global function with example usage: _run_static_range_qat(src_saved_model_path, dst_saved_model_path, quant_opts, signature_def_map) without return types