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 )	MulreturnNonec                  C  sX  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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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
    r3   S/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/mul.pyexport   sJ   z
Mul.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_mul_bcastr   )	r!   r"   r#   r$   r(   r)   r&   r'   r   r1   r3   r3   r4   export_mul_broadcast>   s   zMul.export_mul_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   