o
    ²3Ih;  ã                   @   s    d dl mZ G dd„ deƒZdS )é   )ÚQuantOperatorBasec                       s$   e Zd Z‡ fdd„Zdd„ Z‡  ZS )ÚQArgMaxc                    s   t ƒ  ||¡ d S )N)ÚsuperÚ__init__)ÚselfÚonnx_quantizerÚ	onnx_node©Ú	__class__© úi/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/onnxruntime/quantization/operators/argmax.pyr      s   zQArgMax.__init__c                 C   sX   | j }| j |jd ¡}|d u r| j j|g7  _d S |j|jd< | j j|g7  _d S )Né    )ÚnodeÚ	quantizerÚfind_quantized_valueÚinputÚ	new_nodesÚq_name)r   r   Úquantized_input_valuer   r   r   Úquantize	   s   zQArgMax.quantize)Ú__name__Ú
__module__Ú__qualname__r   r   Ú__classcell__r   r   r	   r   r      s    r   N)Úbase_operatorr   r   r   r   r   r   Ú<module>   s    