U
    rh                     @  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dZeddddZeddddZeddd	d
ZeddddZeddddZ	eddddZ
eddddZdS )BitShiftNone)returnc                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||? }t| ||g|gdd d S )Nr   xyzRIGHTinputsoutputs	direction               Ztest_bitshift_right_uint8r   r   nameonnxhelper	make_nodenparrayastypeZuint8r   noder   r	   r
    r    H/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/bitshift.pyexport_right_unit8   s       zBitShift.export_right_unit8c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||? }t| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r   Ztest_bitshift_right_uint16r   r   r   r   r   r   r   Zuint16r   r   r    r    r!   export_right_unit16   s       zBitShift.export_right_unit16c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||? }t| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r   Ztest_bitshift_right_uint32r   r   r   r   r   r   r   Zuint32r   r   r    r    r!   export_right_unit32$   s       zBitShift.export_right_unit32c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||? }t| ||g|gdd d S )Nr   r   r	   r
   r   r   r   r   r   r   r   Ztest_bitshift_right_uint64r   r   r   r   r   r   r   Zuint64r   r   r    r    r!   export_right_unit64/   s       zBitShift.export_right_unit64c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||> }t| ||g|gdd d S )Nr   r   r	   r
   LEFTr   r   r   r   r   r   Ztest_bitshift_left_uint8r   r   r   r    r    r!   export_left_unit8:   s       zBitShift.export_left_unit8c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||> }t| ||g|gdd d S )Nr   r   r	   r
   r)   r   r   r   r   r   r   Ztest_bitshift_left_uint16r   r#   r   r    r    r!   export_left_unit16E   s       zBitShift.export_left_unit16c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||> }t| ||g|gdd d S )Nr   r   r	   r
   r)   r   r   r   r   r   r   Ztest_bitshift_left_uint32r   r%   r   r    r    r!   export_left_unit32P   s       zBitShift.export_left_unit32c                  C  sl   t jjdddgdgdd} tddd	gtj}td	d
dgtj}||> }t| ||g|gdd d S )Nr   r   r	   r
   r)   r   r   r   r   r   r   Ztest_bitshift_left_uint64r   r'   r   r    r    r!   export_left_unit64[   s       zBitShift.export_left_unit64N)__name__
__module____qualname__staticmethodr"   r$   r&   r(   r*   r+   r,   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   r   r    r    r    r!   <module>   s
   