o
    wi                     @  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ZedddZedddZedd	d
ZedddZedddZ	edddZ
edddZdS )BitShiftreturnNonec                  C  h   t jjdddgdgdd} tg dtj}tg dtj}||? }t| ||g|gd	d
 d S )Nr   xyzRIGHTinputsoutputs	direction         r         test_bitshift_right_uint8r   r   nameonnxhelper	make_nodenparrayastypeuint8r   noder	   r
   r    r%   a/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/bitshift.pyexport_right_unit8      zBitShift.export_right_unit8c                  C  r   )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint16r   r   r   r   r   r    r!   uint16r   r#   r%   r%   r&   export_right_unit16   r(   zBitShift.export_right_unit16c                  C  r   )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint32r   r   r   r   r   r    r!   uint32r   r#   r%   r%   r&   export_right_unit32$   r(   zBitShift.export_right_unit32c                  C  r   )Nr   r	   r
   r   r   r   r   r   test_bitshift_right_uint64r   r   r   r   r   r    r!   uint64r   r#   r%   r%   r&   export_right_unit64/   r(   zBitShift.export_right_unit64c                  C  h   t jjdddgdgdd} tg dtj}tg dtj}||> }t| ||g|gd	d
 d S )Nr   r	   r
   r   LEFTr   r   r   test_bitshift_left_uint8r   r   r#   r%   r%   r&   export_left_unit8:   r(   zBitShift.export_left_unit8c                  C  r5   )Nr   r	   r
   r   r6   r   r   r   test_bitshift_left_uint16r   r*   r#   r%   r%   r&   export_left_unit16E   r(   zBitShift.export_left_unit16c                  C  r5   )Nr   r	   r
   r   r6   r   r   r   test_bitshift_left_uint32r   r.   r#   r%   r%   r&   export_left_unit32P   r(   zBitShift.export_left_unit32c                  C  r5   )Nr   r	   r
   r   r6   r   r   r   test_bitshift_left_uint64r   r2   r#   r%   r%   r&   export_left_unit64[   r(   zBitShift.export_left_unit64N)r   r   )__name__
__module____qualname__staticmethodr'   r,   r0   r4   r8   r:   r<   r>   r%   r%   r%   r&   r      s"    






r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r%   r%   r%   r&   <module>   s   