
    Ng                     J    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 )    N)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returnNc                  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       `/var/www/html/ai-engine/env/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_unit89   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   r9   r   r   r   test_bitshift_left_uint16r   r*   r#   s       r%   export_left_unit16zBitShift.export_left_unit16D       {$$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   r9   r   r   r   test_bitshift_left_uint32r   r0   r#   s       r%   export_left_unit32zBitShift.export_left_unit32O   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   r9   r   r   r   test_bitshift_left_uint64r   r5   r#   s       r%   export_left_unit64zBitShift.export_left_unit64Z   r?   r'   )r   N)__name__
__module____qualname__staticmethodr&   r,   r2   r7   r;   r>   rB   rE    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   )numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rJ   r'   r%   <module>rN      s   
      , , , , , , . . . . . .WS WS WS WS WSt WS WS WS WS WSr'   