o
    ²3IhŸ  ã                   @   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__© új/home/air/sanwanet/gpt-api/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 ©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    