U
    ‰d›  ã                   @   sx   d Z ddlmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ dS )a0  
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.
é    )Úreshape_scaleÚ_InputEqualizationObserverÚ_WeightEqualizationObserverÚcalculate_equalization_scaleÚEqualizationQConfigÚinput_equalization_observerÚweight_equalization_observerÚdefault_equalization_qconfigÚ"fused_module_supports_equalizationÚnn_module_supports_equalizationÚnode_supports_equalizationÚis_equalization_observerÚget_op_node_and_weight_eq_obsÚmaybe_get_weight_eq_obs_nodeÚmaybe_get_next_input_eq_obsÚ!maybe_get_next_equalization_scaleÚscale_input_observerÚscale_weight_nodeÚscale_weight_functionalÚclear_weight_quant_obs_nodeÚremove_nodeÚupdate_obs_for_equalizationÚconvert_eq_obsÚ_convert_equalization_refÚget_layer_sqnr_dictÚget_equalization_qconfig_dictN)Ú__doc__Z"torch.ao.quantization.fx._equalizer   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   © r   r   úC/tmp/pip-unpacked-wheel-ua33x9lu/torch/quantization/fx/_equalize.pyÚ<module>   s   