
    `i                     L    S SK r SSKJr  SSKJrJr  S SK Jr    " S S\5      rg)	    N   )QuantOperatorBase   )QuantizedValueQuantizedValueType)onnx_pbc                   4   ^  \ rS rSrU 4S jrU 4S jrSrU =r$ )GatherQuant
   c                 $   > [         TU ]  X5        g )N)super__init__)selfonnx_quantizer	onnx_node	__class__s      n/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxruntime_tools/quantization/operators/gather.pyr   GatherQuant.__init__   s    3    c                   > U R                   nUR                  S:X  d   eU R                  R                  UR                  S   5      (       d  [
        T	U ]  5         g U R                  R                  US/5      u  p#pEUR                  S   S-   n[        UR                  S   XdS   US   [        R                  5      nXpR                  R                  UR                  S   '   UR                  S   nXaR                  S'   US   UR                  S'   UR                  U5        U R                  =R                  U-  sl        g )NGatherr   
_quantized)nodeop_type	quantizeris_valid_quantize_weightinputr   quantizequantize_inputsoutputr   r   Inputquantized_value_mapappend	new_nodes)
r   r   quantized_input_nameszero_point_namesscale_namesnodesgather_new_outputq_outputgather_original_outputr   s
            r   r   GatherQuant.quantize   s   yy()(77

1FFG NN**4!5 	F	+ !KKN\9 "$++a.2CQR^UefgUh"4":":<=E**4;;q>:!%Q*A-a0

1T  E) r    )__name__
__module____qualname____firstlineno__r   r   __static_attributes____classcell__)r   s   @r   r
   r
   
   s    4* *r   r
   )	onnxbase_operatorr   quant_utilsr   r   r   
onnx_protor
   r-   r   r   <module>r8      s#     , < &
*# *r   