
     NgI                     .    d dl mZ  G d de          ZdS )   )QDQOperatorBasec                   $     e Zd Z fdZd Z xZS )QDQNormalizationc                 L    t                                          ||           d S )N)super__init__)selfonnx_quantizer	onnx_node	__class__s      c/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnxruntime/quantization/operators/norm.pyr   zQDQNormalization.__init__
   s#    33333    c                 6   | j         }|j        dv sJ | j                            |j        d                    | j                            |j        d                   }| j                            |j        d         d|j                  \  }}|r(| j                            |j        d         |           nM|r&| j                            |j        d                    n%| j                            |j        d                    t          |j                  dk    rP|j        d         rC| j        
                    |j        |j        d         |j        d         |j        d                    | j        s$|j        D ]}| j                            |           d S d S )N>   BatchNormalizationLayerNormalizationInstanceNormalization    r   )default_axisop_type)axis   )noder   	quantizerquantize_activation_tensorinputis_input_a_initializeris_tensor_per_channel"quantize_weight_tensor_per_channelquantize_weight_tensorlenquantize_bias_tensornamedisable_qdq_for_node_outputoutput)r	   r   scale_is_initializerscale_is_per_channelscale_channel_axisoutput_names         r   quantizezQDQNormalization.quantize   s   y|ddddd 	11$*Q-@@@  $~DDTZPQ]SS37>3W3WJqM4< 4X 4
 4
00   	EN==djmRd=eeee! 	EN11$*Q-@@@@N55djmDDD tz??Q4:a=N//	4:a=$*UV-Y]YcdeYfggg / 	G#{ G G99+FFFF	G 	GG Gr   )__name__
__module____qualname__r   r)   __classcell__)r   s   @r   r   r   	   sN        4 4 4 4 4G G G G G G Gr   r   N)qdq_base_operatorr   r    r   r   <module>r0      sW    / . . . . .G G G G G G G G G Gr   