quantization_mappings.py 1.1 KB

1234567891011121314151617181920212223242526272829
  1. # flake8: noqa: F401
  2. r"""
  3. This file is in the process of migration to `torch/ao/quantization`, and
  4. is kept here for compatibility while the migration process is ongoing.
  5. If you are adding a new entry/functionality, please, add it to the
  6. `torch/ao/quantization/quantization_mappings.py`, while adding an import statement
  7. here.
  8. """
  9. from torch.ao.quantization.quantization_mappings import (
  10. DEFAULT_REFERENCE_STATIC_QUANT_MODULE_MAPPINGS,
  11. DEFAULT_STATIC_QUANT_MODULE_MAPPINGS,
  12. DEFAULT_QAT_MODULE_MAPPINGS,
  13. DEFAULT_DYNAMIC_QUANT_MODULE_MAPPINGS,
  14. _INCLUDE_QCONFIG_PROPAGATE_LIST,
  15. DEFAULT_FLOAT_TO_QUANTIZED_OPERATOR_MAPPINGS,
  16. DEFAULT_MODULE_TO_ACT_POST_PROCESS,
  17. no_observer_set,
  18. get_default_static_quant_module_mappings,
  19. get_static_quant_module_class,
  20. get_dynamic_quant_module_class,
  21. get_default_qat_module_mappings,
  22. get_default_dynamic_quant_module_mappings,
  23. get_default_qconfig_propagation_list,
  24. get_default_compare_output_module_list,
  25. get_default_float_to_quantized_operator_mappings,
  26. get_quantized_operator,
  27. _get_special_act_post_process,
  28. _has_special_act_post_process,
  29. )