U
    q±Ëh  ã                   @  s4   d dl mZ d dlZd dlmZ G dd„ deƒZdS )é    )ÚannotationsN)ÚOpRunc                   @  s   e Zd Zddd„ZdS )ÚMatMulIntegerNc                 C  sB   |  tj¡}|d k	r||8 }|  tj¡}|d k	r8||8 }|| fS )N)ZastypeÚnpZint32)ÚselfÚAÚBZa_zero_pointZb_zero_pointZA32ZB32© r	   úH/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_matmul_integer.pyÚ_run   s    zMatMulInteger._run)NN)Ú__name__Ú
__module__Ú__qualname__r   r	   r	   r	   r
   r      s   r   )Ú
__future__r   Znumpyr   Zonnx.reference.op_runr   r   r	   r	   r	   r
   Ú<module>   s   