U
    rh                     @  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 )
BitwiseAnd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
bitwiseandinputsoutputs)      Ztest_bitwise_and_i32_2dr   r   namer   r      Ztest_bitwise_and_i16_3d)	onnxhelper	make_noder   npZint32bitwise_andr   Zint16noder	   r
   z r   J/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/bitwiseand.pyexport   s    zBitwiseAnd.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   Z test_bitwise_and_ui64_bcast_3v1dr   )r   r   r      )r   r   r    Ztest_bitwise_and_ui8_bcast_4v3d)	r   r   r   r   r   Zuint64r   r   Zuint8r   r   r   r   export_bitwiseand_broadcast#   s$       z&BitwiseAnd.export_bitwiseand_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   