o
    wiZ                     @  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  s  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	 d S )Nr   abc)inputsoutputs      test_matmul_2d)r   r   name   test_matmul_3d   test_matmul_4d)
onnxhelper	make_nodenprandomrandnastypefloat32matmulr   )noder   r	   r
    r   _/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/matmul.pyexport   s"   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   