U
    Ç9%eâ  ã                   @   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   ú^/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/quantization/fx/_equalize.pyÚ<module>   s   