U
    “±Ëh;  ã                   @   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__)ÚselfZonnx_quantizerZ	onnx_node©Ú	__class__© úM/tmp/pip-unpacked-wheel-socb9apf/onnxruntime/quantization/operators/argmax.pyr      s    zQArgMax.__init__c                 C   sX   | j }| j |jd ¡}|d kr6| j j|g7  _d S |j|jd< | j j|g7  _d S )Né    )ÚnodeZ	quantizerZfind_quantized_valueÚinputZ	new_nodesZq_name)r   r   Zquantized_input_valuer	   r	   r
   Úquantize	   s    zQArgMax.quantize)Ú__name__Ú
__module__Ú__qualname__r   r   Ú__classcell__r	   r	   r   r
   r      s   r   N)Zbase_operatorr   r   r	   r	   r	   r
   Ú<module>   s   