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 )ÚQResizec                    s   t ƒ  ||¡ d S ©N©ÚsuperÚ__init__©ÚselfZonnx_quantizerZ	onnx_node©Ú	__class__© úM/tmp/pip-unpacked-wheel-socb9apf/onnxruntime/quantization/operators/resize.pyr      s    zQResize.__init__c                    s<   | j }|jdkst‚| jjdk r2tt| ƒ ¡  d S tƒ  ¡ S ©NZResizeé   )ÚnodeÚop_typeÚAssertionErrorÚ	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      s   r   c                       s(   e Zd Z‡ fdd„Z‡ fdd„Z‡  ZS )Ú	QDQResizec                    s   t ƒ  ||¡ d S r   r   r	   r   r   r   r      s    zQDQResize.__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QDQResize.quantizer   r   r   r   r   r      s   r   N)Z	direct_q8r   r   r   r   r   r   r   r   Ú<module>   s   