o
    so™iø  ã                   @  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dS )ÚMatMulÚreturnÚNonec                  C  sP  t jjdddgdgd} tj dd¡ tj¡}tj dd¡ tj¡}t ||¡}t	| ||g|gdd	 tj d
dd¡ tj¡}tj d
dd¡ tj¡}t ||¡}t	| ||g|gdd	 tj dd
dd¡ tj¡}tj dd
dd¡ tj¡}t ||¡}t	| ||g|gdd	 tj dddd¡ tj¡}tj dd
dd
¡ tj¡}t ||¡}t	| ||g|gdd	 tj d¡ tj¡}tj d
dd¡ tj¡}t ||¡}t	| ||g|gdd	 tj dd
dd¡ tj¡}tj d¡ tj¡}t ||¡}t	| ||g|gdd	 tj d¡ tj¡}tj d¡ tj¡}t ||¡}t	| ||g|gdd	 d S )Nr   ÚaÚbÚc)ÚinputsÚoutputsé   é   Útest_matmul_2d)r   r   Únameé   Útest_matmul_3dé   Útest_matmul_4dÚtest_matmul_bcastÚtest_matmul_1d_3dÚtest_matmul_4d_1dÚtest_matmul_1d_1d)
ÚonnxÚhelperÚ	make_nodeÚnpÚrandomÚrandnÚastypeÚfloat32Úmatmulr   )Únoder   r	   r
   © r#   úV/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/matmul.pyÚexport   sB   ýzMatMul.exportN)r   r   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr%   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   