U
    U?h                     @   s4   d dl mZmZ G dd deZG dd deZdS )   )Direct8BitOpQDQDirect8BitOpc                       s(   e Zd Z fddZ fddZ  ZS )QMaxPoolc                    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/maxpool.pyr      s    zQMaxPool.__init__c                    s<   | j }|jdkst| jjdk r2tt|   d S t  S NZMaxPool   )nodeop_typeAssertionError	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      s   r   c                       s(   e Zd Z fddZ fddZ  ZS )
QDQMaxPoolc                    s   t  || d S r   r   r	   r   r   r   r      s    zQDQMaxPool.__init__c                    s.   | j }|jdkst| jjdk r$d S t  S r   )r   r   r   r   r   r   r   r   r   r   r   r      s
    zQDQMaxPool.quantizer   r   r   r   r   r      s   r   N)Z	direct_q8r   r   r   r   r   r   r   r   <module>   s   