o
    soi                     @  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 )	
BitwiseAnd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
bitwiseandinputsoutputs)      test_bitwise_and_i32_2dr   r   namer   r      test_bitwise_and_i16_3d)	onnxhelper	make_noder   npint32bitwise_andr   int16noder	   r
   z r!   Z/home/ubuntu/.local/lib/python3.10/site-packages/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    test_bitwise_and_ui64_bcast_3v1dr   )r   r   r      )r   r   r%   test_bitwise_and_ui8_bcast_4v3d)	r   r   r   r   r   uint64r   r   uint8r   r!   r!   r"   export_bitwiseand_broadcast#   s   z&BitwiseAnd.export_bitwiseand_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   