U
    r±Ëh²	  ã                   @  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                   @  s0   e Zd Zeddœdd„ƒZeddœdd„ƒZdS )ÚOrÚNone)Úreturnc                  C  s  t jjdddgdgd} tj dd¡dk t¡}tj dd¡dk t¡}t ||¡}t	| ||g|gd	d
 tj ddd¡dk t¡}tj ddd¡dk t¡}t ||¡}t	| ||g|gdd
 tj dddd¡dk t¡}tj dddd¡dk t¡}t ||¡}t	| ||g|gdd
 d S )Nr   ÚxÚyÚor©ÚinputsÚoutputsé   é   r   Z	test_or2d©r   r   Únameé   Z	test_or3dé   Z	test_or4d©
ÚonnxÚhelperZ	make_nodeÚnpÚrandomZrandnZastypeÚboolÚ
logical_orr   ©Únoder   r	   Úz© r   úB/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/or.pyÚexport   s"    ýz	Or.exportc                  C  sÊ  t jjdddgdgd} tj ddd¡d	k t¡}tj d¡d	k t¡}t ||¡}t	| ||g|gd
d tj ddd¡d	k t¡}tj dd¡d	k t¡}t ||¡}t	| ||g|gdd tj dddd¡d	k t¡}tj dd¡d	k t¡}t ||¡}t	| ||g|gdd tj dddd¡d	k t¡}tj ddd¡d	k t¡}t ||¡}t	| ||g|gdd tj dddd¡d	k t¡}tj dddd¡d	k t¡}t ||¡}t	| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   Ztest_or_bcast3v1dr   Ztest_or_bcast3v2dr   Ztest_or_bcast4v2dZtest_or_bcast4v3dé   Ztest_or_bcast4v4dr   r   r   r   r   Úexport_or_broadcast(   s2    ýzOr.export_or_broadcastN)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr    r"   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
   