utils.py 791 B

123456789101112131415
  1. # flake8: noqa: F401
  2. r"""Quantized Reference Modules
  3. This module is in the process of migration to
  4. `torch/ao/nn/quantized/reference`, and is kept here for
  5. compatibility while the migration process is ongoing.
  6. If you are adding a new entry/functionality, please, add it to the
  7. appropriate file under the `torch/ao/nn/quantized/reference`,
  8. while adding an import statement here.
  9. """
  10. from torch.ao.nn.quantized.reference.modules.utils import _quantize_weight
  11. from torch.ao.nn.quantized.reference.modules.utils import _quantize_and_dequantize_weight
  12. from torch.ao.nn.quantized.reference.modules.utils import _save_weight_qparams
  13. from torch.ao.nn.quantized.reference.modules.utils import _get_weight_qparam_keys
  14. from torch.ao.nn.quantized.reference.modules.utils import ReferenceQuantizedModule