
    /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 )MulreturnNonec                    t           j                            dddgdg          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z  }t          | ||g|gd	           t          j        	                    d
dd                              t          j                  }t          j        	                    d
dd                              t          j                  }||z  }t          | ||g|gd	           t          j        
                    ddt          j                  }t          j        
                    ddt          j                  }||z  }t          | ||g|gd	           t          j        
                    ddt          j                  }t          j        
                    ddt          j                  }||z  }t          | ||g|gd	           t          j        
                    ddt          j                  }t          j        
                    ddt          j                  }||z  }t          | ||g|gd	           t          j        
                    ddt          j                  }t          j        
                    ddt          j                  }||z  }t          | ||g|gd	           t          j        
                    ddt          j                  }t          j        
                    ddt          j                  }||z  }t          | ||g|gd	           t          j        
                    ddt          j                  }t          j        
                    ddt          j                  }||z  }t          | ||g|gd	           d S )Nr   xyzinputsoutputs)         )         test_mul_exampler   r   namer   r   r   test_mul)r   r   r   )sizedtype   test_mul_int8test_mul_int16test_mul_uint8test_mul_uint16test_mul_uint32test_mul_uint64)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandintint8int16uint8uint16uint32uint64noder   r   r   s       d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/mul.pyexportz
Mul.export   sg   {$$:E % 
 
 HYYY&&rz22HYYY&&rz22EtQFQC6HIIIIIOOAq!$$++BJ77IOOAq!$$++BJ77EtQFQCjAAAAIairw??Iby@@EtQFQCoFFFFIairx@@IbyAAEtQFQC6FGGGGIairx@@IbyAAEtQFQC6FGGGGIairyAAIby	BBEtQFQC6GHHHHIairyAAIby	BBEtQFQC6GHHHHIairyAAIby	BBEtQFQC6GHHHHHH    c                 z   t           j                            dddgdg          } t          j                            ddd                              t          j                  }t          j                            d                              t          j                  }||z  }t          | ||g|gd	
           d S )Nr   r   r   r   r   r   r   r   test_mul_bcastr   )	r$   r%   r&   r'   r+   r,   r)   r*   r   r4   s       r6   export_mul_broadcastzMul.export_mul_broadcast>   s    {$$:E % 
 
 IOOAq!$$++BJ77IOOA%%bj11EtQFQC6FGGGGGGr8   N)r   r	   )__name__
__module____qualname__staticmethodr7   r;    r8   r6   r   r      s]        -I -I -I \-I^ 
H 
H 
H \
H 
H 
Hr8   r   )

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