U
    qhH                     @  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 )IsInfNc                 C  sJ   |r |rt |fS t |fS |r0t |fS t j|jt jdd}|fS )NF)ZdtypeZ
fill_value)npisinfZisneginfZisposinffullshapeZbool_)selfdataZdetect_negativeZdetect_positiveres r   ?/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_isinf.py_run   s    z
IsInf._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   