
    /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 )AddreturnNonec                    t           j                            dddgdg          } t          j                            ddd                              t          j                  }t          j                            ddd                              t          j                  }t          | ||g||z   gd	
           t          j        	                    ddt          j
                  }t          j        	                    ddt          j
                  }t          | ||g||z   gd
           t          j        	                    ddt          j                  }t          j        	                    ddt          j                  }t          | ||g||z   gd
           t          j        	                    ddt          j                  }t          j        	                    ddt          j                  }t          | ||g||z   gd
           t          j        	                    ddt          j                  }t          j        	                    ddt          j                  }t          | ||g||z   gd
           t          j        	                    ddt          j                  }t          j        	                    ddt          j                  }t          | ||g||z   gd
           t          j        	                    ddt          j                  }t          j        	                    ddt          j                  }t          | ||g||z   g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   s      d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/add.pyexportz
Add.export   s   {$$:G % 
 
 IOOAq!$$++BJ77IOOAq!$$++BJ77tQFQUG*EEEEIby@@Iby@@tQFQUG/JJJJIbyAAIbyAAtQFQUG:JKKKKIbyAAIbyAAtQFQUG:JKKKKIby	BBIby	BBtQFQUG:KLLLLIby	BBIby	BBtQFQUG:KLLLLIby	BBIby	BBtQFQUG:KLLLLLL    c                 v   t           j                            dddgdg          } t          j                            ddd                              t          j                  }t          j                            d                              t          j                  }t          | ||g||z   gd	
           d S )Nr   r   r   r   r   r   r   r   test_add_bcastr   )	r    r!   r"   r#   r$   r%   r&   r'   r   r/   s      r1   export_add_broadcastzAdd.export_add_broadcast2   s    {$$:G % 
 
 IOOAq!$$++BJ77IOOA%%bj11tQFQUG:JKKKKKKr3   N)r   r	   )__name__
__module____qualname__staticmethodr2   r6    r3   r1   r   r      s]        !M !M !M \!MF 	L 	L 	L \	L 	L 	Lr3   r   )

__future__r   numpyr#   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r;   r3   r1   <module>r@      s    # " " " " "      , , , , , , . . . . . ./L /L /L /L /L$ /L /L /L /L /Lr3   