o
    ¢Äi;  ã                   @   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__© úc/home/kim/smarthome/.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é    )Ú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    