
    /j                    V    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e          Z	dS )    )annotationsN)Base)expectc                      e Zd Zedd            Zedd            Zedd            Zedd            Zedd            Zedd            Z	edd	            Z
edd
            ZdS )BitShiftreturnNonec                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z	  }t          | ||g|g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   s       i/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/bitshift.pyexport_right_unit8zBitShift.export_right_unit8   s    {$$SzC5G % 
 
 HZZZ  ''11HYYY&&rx00FtQFQC6QRRRRRR    c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z	  }t          | ||g|gd	
           d S )Nr   r   r   r   r   r   r   r   test_bitshift_right_uint16r   r   r   r    r!   r"   r#   uint16r   r%   s       r'   export_right_unit16zBitShift.export_right_unit16       {$$SzC5G % 
 
 HZZZ  ''	22HYYY&&ry11FtQFQC6RSSSSSSr)   c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z	  }t          | ||g|gd	
           d S )Nr   r   r   r   r   r   r   r   test_bitshift_right_uint32r   r   r   r    r!   r"   r#   uint32r   r%   s       r'   export_right_unit32zBitShift.export_right_unit32$   r/   r)   c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z	  }t          | ||g|gd	
           d S )Nr   r   r   r   r   r   r   r   test_bitshift_right_uint64r   r   r   r    r!   r"   r#   uint64r   r%   s       r'   export_right_unit64zBitShift.export_right_unit64/   r/   r)   c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z  }t          | ||g|gd	
           d S )Nr   r   r   r   LEFTr   r   r   test_bitshift_left_uint8r   r   r%   s       r'   export_left_unit8zBitShift.export_left_unit8:   s    {$$SzC5F % 
 
 HZZZ  ''11HYYY&&rx00FtQFQC6PQQQQQQr)   c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z  }t          | ||g|gd	
           d S )Nr   r   r   r   r;   r   r   r   test_bitshift_left_uint16r   r,   r%   s       r'   export_left_unit16zBitShift.export_left_unit16E       {$$SzC5F % 
 
 HZZZ  ''	22HYYY&&ry11FtQFQC6QRRRRRRr)   c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z  }t          | ||g|gd	
           d S )Nr   r   r   r   r;   r   r   r   test_bitshift_left_uint32r   r2   r%   s       r'   export_left_unit32zBitShift.export_left_unit32P   rA   r)   c                 T   t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z  }t          | ||g|gd	
           d S )Nr   r   r   r   r;   r   r   r   test_bitshift_left_uint64r   r7   r%   s       r'   export_left_unit64zBitShift.export_left_unit64[   rA   r)   N)r   r	   )__name__
__module____qualname__staticmethodr(   r.   r4   r9   r=   r@   rD   rG    r)   r'   r   r      s4       S S S \S T T T \T T T T \T T T T \T R R R \R S S S \S S S S \S S S S \S S Sr)   r   )

__future__r   numpyr!   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rL   r)   r'   <module>rQ      s    # " " " " "      , , , , , , . . . . . .WS WS WS WS WSt WS WS WS WS WSr)   