
    bIg                    N    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	y)    )annotationsN)Base)expectc                  0    e Zd Zedd       Zedd       Zy)Subc                 b   t         j                  j                  dddgdg      } t        j                  g d      j                  t        j                        }t        j                  g d      j                  t        j                        }||z
  }t        | ||g|gd	       t        j                  j                  d
dd      j                  t        j                        }t        j                  j                  d
dd      j                  t        j                        }||z
  }t        | ||g|gd	       t        j                  j                  dddt        j                        }t        j                  j                  ddt        j                        }||z
  }t        | ||g|gd	       y )Nr   xyzinputsoutputs)         )r   r   r   test_sub_exampler   r   namer         test_sub      )r   r   r   )sizedtypetest_sub_uint8)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandintuint8noder	   r
   r   s       \/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/sub.pyexportz
Sub.export   sI   {{$$:E % 
 HHY&&rzz2HHY&&rzz2EtQFQC6HIIIOOAq!$++BJJ7IIOOAq!$++BJJ7EtQFQCjAIIb"9BHHEIIbyAEtQFQC6FG    c                 p   t         j                  j                  dddgdg      } t        j                  j                  ddd      j                  t        j                        }t        j                  j                  d      j                  t        j                        }||z
  }t        | ||g|gd	
       y )Nr   r	   r
   r   r   r   r   r   test_sub_bcastr   )	r   r   r   r    r$   r%   r"   r#   r   r(   s       r*   export_sub_broadcastzSub.export_sub_broadcast%   s    {{$$:E % 
 IIOOAq!$++BJJ7IIOOA%%bjj1EtQFQC6FGr,   N)returnNone)__name__
__module____qualname__staticmethodr+   r/    r,   r*   r   r      s,    H H, 
H 
Hr,   r   )

__future__r   numpyr    r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r6   r,   r*   <module>r;      s$    #   , .#H$ #Hr,   