o
    
j;                     @   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__ \/home/kuhnn/.local/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    