U
    rhZ                     @  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dZdS )MatMulNone)returnc                  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      Ztest_matmul_2d)r   r   name   Ztest_matmul_3d   Ztest_matmul_4d)
onnxhelperZ	make_nodenprandomZrandnZastypeZfloat32matmulr   )noder   r	   r
    r   F/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/matmul.pyexport   s"    zMatMul.exportN)__name__
__module____qualname__staticmethodr   r   r   r   r   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s
   