
    /j                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de          Z	dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )IsNaNreturnNonec                 >   t           j                            ddgdg          } t          j        dt          j        t          j        dt          j         t          j        gt          j                  }t          j        |          }t          | |g|gd	           d S )
Nr   xyinputsoutputs333333ffffff@dtype
test_isnanr   r   name)
onnxhelper	make_nodenparraynaninffloat32isnanr   noder   r   s      f/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/isnan.pyexportzIsNaN.export   s    {$$5E % 
 
 HdBFBFC"&"&ATTTHQKKtQC!<@@@@@@    c                 >   t           j                            ddgdg          } t          j        dt          j        t          j        dt          j         t          j        gt          j                  }t          j        |          }t          | |g|gd	           d S )
Nr   r   r   r   r   r   r   test_isnan_float16r   )
r   r   r   r   r   r   r   float16r   r   r    s      r"   export_float16zIsNaN.export_float16   s    {$$5E % 
 
 HdBFBFC"&"&ATTTHQKKtQC!3GHHHHHHr$   N)r   r	   )__name__
__module____qualname__staticmethodr#   r(    r$   r"   r   r      s\        	A 	A 	A \	A 	I 	I 	I \	I 	I 	Ir$   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r-   r$   r"   <module>r2      s    # " " " " "      , , , , , , . . . . . .I I I I ID I I I I Ir$   