o
    wi,
                     @  sH   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 deZ	dS )    )annotationsN)Base)expectc                   @  s(   e Zd ZedddZedddZdS )	SubreturnNonec                  C  sd  t jjdddgdgd} tg dtj}tg dtj}|| }t| ||g|gdd	 tj	d
ddtj}tj	d
ddtj}|| }t| ||g|gdd	 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd	 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd	 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd	 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd	 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd	 tjj
dddtjd}tjj
ddtjd}|| }t| ||g|gdd	 d S )Nr   xyzinputsoutputs)         )r   r   r   test_sub_exampler   r   namer         test_sub      )r   r   r   )sizedtypetest_sub_int8test_sub_int16test_sub_uint8test_sub_uint16test_sub_uint32test_sub_uint64)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandintint8int16uint8uint16uint32uint64noder   r	   r
    r3   \/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/sub.pyexport   sJ   z
Sub.exportc                  C  sf   t jjdddgdgd} tjdddtj}tjdtj}|| }t| ||g|gd	d
 d S )Nr   r   r	   r
   r   r   r   r   test_sub_bcastr   )	r!   r"   r#   r$   r(   r)   r&   r'   r   r1   r3   r3   r4   export_sub_broadcast>   s   zSub.export_sub_broadcastN)r   r   )__name__
__module____qualname__staticmethodr5   r7   r3   r3   r3   r4   r      s
    /r   )

__future__r   numpyr$   r!   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r3   r3   r3   r4   <module>   s   