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 )ÚQResizec                    ó   t ƒ  ||¡ d S ©N©ÚsuperÚ__init__©ÚselfÚonnx_quantizerÚ	onnx_node©Ú	__class__© ú]/home/ubuntu/.local/lib/python3.10/site-packages/onnxruntime/quantization/operators/resize.pyr
      ó   zQResize.__init__c                    s<   | j }|jdks
J ‚| jjdk rtt| ƒ ¡  d S tƒ  ¡ S ©NÚResizeé   )ÚnodeÚop_typeÚ	quantizerÚopset_versionr	   r   Úquantize©r   r   r   r   r   r      s   
zQResize.quantize©Ú__name__Ú
__module__Ú__qualname__r
   r   Ú__classcell__r   r   r   r   r      ó    r   c                       r   )Ú	QDQResizec                    r   r   r   r   r   r   r   r
      r   zQDQResize.__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QDQResize.quantizer   r   r   r   r   r#      r"   r#   N)Ú	direct_q8r   r   r   r#   r   r   r   r   Ú<module>   s    