
    h;                     &    S SK Jr   " S S\5      rg)   )QuantOperatorBasec                   .   ^  \ rS rSrU 4S jrS rSrU =r$ )QArgMax   c                 $   > [         TU ]  X5        g )N)super__init__)selfonnx_quantizer	onnx_node	__class__s      c/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/quantization/operators/argmax.pyr	   QArgMax.__init__   s    3    c                 &   U R                   nU R                  R                  UR                  S   5      nUc!  U R                  =R                  U/-  sl        g UR
                  UR                  S'   U R                  =R                  U/-  sl        g )N    )node	quantizerfind_quantized_valueinput	new_nodesq_name)r
   r   quantized_input_values      r   quantizeQArgMax.quantize	   sr    yy $ C CDJJqM R (NN$$.$-44

1  TF* r    )__name__
__module____qualname____firstlineno__r	   r   __static_attributes____classcell__)r   s   @r   r   r      s    4	+ 	+r   r   N)base_operatorr   r   r   r   r   <module>r$      s    ,+ +r   