U
    rhx                     @  sT   d dl mZ d dlZd dlZd dlmZ d dlmZ d dl	m
Z
 G dd deZdS )    )annotationsN)Base)expect)create_random_intc                   @  s0   e Zd ZeddddZeddddZdS )	BitwiseOrNone)returnc                  C  s   t jjdddgdgd} tdtj}tdtj}t||}t| ||g|gdd td	tj}td	tj}t||}t| ||g|gd
d d S )Nr   xy	bitwiseorinputsoutputs)      Ztest_bitwise_or_i32_2dr   r   namer   r         Ztest_bitwise_or_i16_4d)	onnxhelper	make_noder   npZint32
bitwise_orr   Zint8noder	   r
   z r   I/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/bitwiseor.pyexport   s    zBitwiseOr.exportc                  C  s   t jjdddgdgd} tdtj}tdtj}t||}t| ||g|gdd	 td
tj}tdtj}t||}t| ||g|gdd	 d S )Nr   r	   r
   r   r   )r   r   r   )r   Ztest_bitwise_or_ui64_bcast_3v1dr   r   )r   r   r   Ztest_bitwise_or_ui8_bcast_4v3d)	r   r   r   r   r   Zuint64r   r   Zuint8r   r   r   r   export_bitwiseor_broadcast"   s    z$BitwiseOr.export_bitwiseor_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   Zonnx.numpy_helperr   r   r   r   r   r   <module>   s   