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eddd„ƒZdS )	ÚOrÚreturnÚNonec                  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   Ú	test_or2d©r   r   Únameé   Ú	test_or3dé   Ú	test_or4d©
ÚonnxÚhelperÚ	make_nodeÚnpÚrandomÚrandnÚastypeÚboolÚ
logical_orr   ©Únoder   r	   Úz© r$   úR/home/ubuntu/.local/lib/python3.10/site-packages/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   Útest_or_bcast3v1dr   Útest_or_bcast3v2dr   Útest_or_bcast4v2dÚtest_or_bcast4v3dé   Útest_or_bcast4v4dr   r!   r$   r$   r%   Úexport_or_broadcast(   s2   ýzOr.export_or_broadcastN)r   r   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr&   r-   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   