
    Si                    N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )annotationsN)Base)expectc                  d    \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       rSr	g)	IsInf   c                 P   [         R                  R                  SS/S/S9n [        R                  " S[        R
                  [        R                  S[        R                  * [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/SS	9  g )
Nr   xyinputsoutputs333333ffffff@dtype
test_isinfr   r   name)
onnxhelper	make_nodenparraynaninffloat32isinfr   noder
   r   s      `/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/isinf.pyexport_infinityIsInf.export_infinity   sz    {{$$5E % 
 HHdBFFBFFC"&&"&&ATHHQKtC!<@    c                 R   [         R                  R                  SS/S/SS9n [        R                  " S[        R
                  [        R                  S[        R                  * [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/S	S
9  g )Nr   r
   r   r   )r   r   detect_negative333333g@r   test_isinf_positiver   )
r   r   r   r   r   r   r   r   isposinfr   r   s      r!   export_positive_infinity_only#IsInf.export_positive_infinity_only   s{    {{$$SEC5! % 
 HHdBFFBFFC"&&"&&ATKKNtC!3HIr$   c                 R   [         R                  R                  SS/S/SS9n [        R                  " S[        R
                  [        R                  S[        R                  * [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/S	S
9  g )Nr   r
   r   r   )r   r   detect_positiver'   gr   test_isinf_negativer   )
r   r   r   r   r   r   r   r   isneginfr   r   s      r!   export_negative_infinity_only#IsInf.export_negative_infinity_only$   s{    {{$$SEC5! % 
 HHdBFFBFFD266'266B"**UKKNtC!3HIr$   c                 P   [         R                  R                  SS/S/S9n [        R                  " S[        R
                  [        R                  S[        R                  * [        R                  /[        R                  S9n[        R                  " U5      n[        X/U/SS	9  g )
Nr   r
   r   r   r   r   r   test_isinf_float16r   )
r   r   r   r   r   r   r   float16r   r   r   s      r!   export_infinity_float16IsInf.export_infinity_float16.   s{    {{$$5E % 
 HHdBFFBFFC"&&"&&ATHHQKtC!3GHr$    N)returnNone)
__name__
__module____qualname____firstlineno__staticmethodr"   r*   r0   r5   __static_attributes__r7   r$   r!   r   r      sX    	A 	A J J J J 	I 	Ir$   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r7   r$   r!   <module>rD      s$    #   , .+ID +Ir$   