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__© úN/tmp/pip-unpacked-wheel-socb9apf/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   