o
    
j                     @   s4   d dl mZmZ G dd deZG dd deZdS )   )Direct8BitOpQDQDirect8BitOpc                       (   e Zd Z fddZ fddZ  ZS )QMaxPoolc                       t  || d S Nsuper__init__selfonnx_quantizer	onnx_node	__class__ ]/home/kuhnn/.local/lib/python3.10/site-packages/onnxruntime/quantization/operators/maxpool.pyr
         zQMaxPool.__init__c                    s<   | j }|jdks
J | jjdk rtt|   d S t  S NMaxPool   )nodeop_type	quantizeropset_versionr	   r   quantizer   r   r   r   r   r      s   
zQMaxPool.quantize__name__
__module____qualname__r
   r   __classcell__r   r   r   r   r          r   c                       r   )
QDQMaxPoolc                    r   r   r   r   r   r   r   r
      r   zQDQMaxPool.__init__c                    s.   | j }|jdks
J | jjdk rd S t  S r   )r   r   r   r   r	   r   r   r   r   r   r      s
   
zQDQMaxPool.quantizer   r   r   r   r   r#      r"   r#   N)	direct_q8r   r   r   r#   r   r   r   r   <module>   s    