
    g;                     &    d dl mZ  G d de      Zy)   )QuantOperatorBasec                   $     e Zd Z fdZd Z xZS )QArgMaxc                 &    t         |   ||       y )N)super__init__)selfonnx_quantizer	onnx_node	__class__s      ^/var/www/openai/venv/lib/python3.12/site-packages/onnxruntime/quantization/operators/argmax.pyr   zQArgMax.__init__   s    3    c                 $   | j                   }| j                  j                  |j                  d         }|!| j                  xj                  |gz  c_        y |j
                  |j                  d<   | j                  xj                  |gz  c_        y )N    )node	quantizerfind_quantized_valueinput	new_nodesq_name)r	   r   quantized_input_values      r   quantizezQArgMax.quantize	   sr    yy $ C CDJJqM R (NN$$.$-44

1  TF* r   )__name__
__module____qualname__r   r   __classcell__)r   s   @r   r   r      s    4	+r   r   N)base_operatorr   r    r   r   <module>r      s    ,+ +r   