o
    soi                     @  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 )	AddreturnNonec                  C  s  t jjdddgdg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tj
d}tjj	ddtj
d}t| ||g|| gdd
 tjj	ddtjd}tjj	ddtjd}t| ||g|| gdd
 tjj	ddtjd}tjj	ddtjd}t| ||g|| gdd
 tjj	ddtjd}tjj	ddtjd}t| ||g|| gdd
 tjj	ddtjd}tjj	ddtjd}t| ||g|| gdd
 tjj	ddtjd}tjj	ddtjd}t| ||g|| gdd
 d S )Nr   xysuminputsoutputs         test_addr   r   name   )r   r   r   )sizedtypetest_add_int8test_add_int16test_add_uint8test_add_uint16test_add_uint32test_add_uint64)onnxhelper	make_nodenprandomrandnastypefloat32r   randintint8int16uint8uint16uint32uint64noder   r	    r.   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/add.pyexport   s4   z
Add.exportc                  C  sb   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_add_bcastr   )	r   r   r   r    r!   r"   r#   r$   r   r,   r.   r.   r/   export_add_broadcast2   s   zAdd.export_add_broadcastN)r   r   )__name__
__module____qualname__staticmethodr0   r2   r.   r.   r.   r/   r      s
    #r   )

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