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 )ÚModNc                 C  s^   |p| j }|dkr t  ||¡fS |jtjtjtjfkrJt t  ||¡¡fS t t ||¡¡fS )Né   )ÚfmodÚnpZdtypeZfloat16Zfloat32Zfloat64Z
nan_to_numÚmod)ÚselfÚaÚbr   © r   ú=/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_mod.pyÚ_run   s    
zMod._run)N)Ú__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   