| # 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 |
| appropriate files under `torch/ao/quantization/fx/`, while adding an import statement |
| here. |
| """ |
| from torch.ao.quantization.fx._equalize import ( |
| _convert_equalization_ref, |
| _InputEqualizationObserver, |
| _WeightEqualizationObserver, |
| calculate_equalization_scale, |
| clear_weight_quant_obs_node, |
| convert_eq_obs, |
| CUSTOM_MODULE_SUPP_LIST, |
| custom_module_supports_equalization, |
| default_equalization_qconfig, |
| EqualizationQConfig, |
| fused_module_supports_equalization, |
| get_equalization_qconfig_dict, |
| get_layer_sqnr_dict, |
| get_op_node_and_weight_eq_obs, |
| input_equalization_observer, |
| is_equalization_observer, |
| maybe_get_next_equalization_scale, |
| maybe_get_next_input_eq_obs, |
| maybe_get_weight_eq_obs_node, |
| nn_module_supports_equalization, |
| node_supports_equalization, |
| remove_node, |
| reshape_scale, |
| scale_input_observer, |
| scale_weight_functional, |
| scale_weight_node, |
| update_obs_for_equalization, |
| weight_equalization_observer, |
| ) |
