o
    ¢ÄiŸ  ã                   @   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__©ÚselfZonnx_quantizerZ	onnx_node©Ú	__class__© úd/home/kim/smarthome/.venv/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 ©NZ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)Z	direct_q8r   r   r   r    r   r   r   r   Ú<module>   s    