o
    soih                     @  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                   @  s(   e Zd ZedddZedddZdS )		BitwiseOrreturnNonec                  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)      test_bitwise_or_i32_2dr   r   namer   r         test_bitwise_or_i16_4d)	onnxhelper	make_noder   npint32
bitwise_orr   int8noder	   r
   z r"   Y/home/ubuntu/.local/lib/python3.10/site-packages/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   test_bitwise_or_ui64_bcast_3v1dr   r   )r   r   r   test_bitwise_or_ui8_bcast_4v3d)	r   r   r   r   r   uint64r   r   uint8r   r"   r"   r#   export_bitwiseor_broadcast"   s   z$BitwiseOr.export_bitwiseor_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   onnx.numpy_helperr   r   r"   r"   r"   r#   <module>   s   