U
    Mh                     @   s   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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.
    )_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_observerN)__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    r   r   Q/var/www/html/venv/lib/python3.8/site-packages/torch/quantization/fx/_equalize.py<module>   s   