U
    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__ [/var/www/html/venv/lib/python3.8/site-packages/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   