mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8a094e3270
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64985 moving quantization_mappings.py and fuser_method_mappings.py to the ao folder while retaining backwards compatibility also added dict test ghstack-source-id: 138215312 Test Plan: buck test mode/dev //caffe2/test:quantization https://www.internalfb.com/intern/testinfra/testrun/7036874471986444 buck test mode/dev //caffe2/test:quantization -- TestAOMigrationQuantization https://www.internalfb.com/intern/testinfra/testrun/5348024625792701 Reviewed By: z-a-f Differential Revision: D30982551 fbshipit-source-id: 00f53bd44009d6012a7de852000aad6885131edb
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
# flake8: noqa: F401
|
|
r"""
|
|
This file is in the process of migration to `torch/ao/quantization`, and
|
|
is kept here for compatibility while the migration process is ongoing.
|
|
If you are adding a new entry/functionality, please, add it to the
|
|
`torch/ao/quantization/quantization_mappings.py`, while adding an import statement
|
|
here.
|
|
"""
|
|
from torch.ao.quantization.quantization_mappings import (
|
|
DEFAULT_REFERENCE_STATIC_QUANT_MODULE_MAPPINGS,
|
|
DEFAULT_STATIC_QUANT_MODULE_MAPPINGS,
|
|
DEFAULT_QAT_MODULE_MAPPINGS,
|
|
DEFAULT_DYNAMIC_QUANT_MODULE_MAPPINGS,
|
|
_INCLUDE_QCONFIG_PROPAGATE_LIST,
|
|
DEFAULT_FLOAT_TO_QUANTIZED_OPERATOR_MAPPINGS,
|
|
DEFAULT_MODULE_TO_ACT_POST_PROCESS,
|
|
no_observer_set,
|
|
get_default_static_quant_module_mappings,
|
|
get_static_quant_module_class,
|
|
get_dynamic_quant_module_class,
|
|
get_default_qat_module_mappings,
|
|
get_default_dynamic_quant_module_mappings,
|
|
get_default_qconfig_propagation_list,
|
|
get_default_compare_output_module_list,
|
|
get_default_float_to_quantized_operator_mappings,
|
|
get_quantized_operator,
|
|
_get_special_act_post_process,
|
|
_has_special_act_post_process,
|
|
)
|