
    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dS )SubreturnNc                  h   t           j                            dddgdg          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }||z
  }t          | ||g|gd	           t          j        	                    d
dd                              t          j                  }t          j        	                    d
dd                              t          j                  }||z
  }t          | ||g|gd	           t          j        
                    dddt          j                  }t          j        
                    ddt          j                  }||z
  }t          | ||g|gd	           d S )Nr   xyzinputsoutputs)         )r   r   r   test_sub_exampler   r   namer         test_sub      )r   r   r   )sizedtypetest_sub_uint8)onnxhelper	make_nodenparrayastypefloat32r   randomrandnrandintuint8noder	   r
   r   s       [/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/sub.pyexportz
Sub.export   s{   {$$:E % 
 
 HYYY&&rz22HYYY&&rz22EtQFQC6HIIIIIOOAq!$$++BJ77IOOAq!$$++BJ77EtQFQCjAAAAIb"9BHEEIbyAAEtQFQC6FGGGGGG    c                  z   t           j                            dddgdg          } t          j                            ddd                              t          j                  }t          j                            d                              t          j                  }||z
  }t          | ||g|gd	
           d S )Nr   r	   r
   r   r   r   r   r   test_sub_bcastr   )	r   r   r   r    r$   r%   r"   r#   r   r(   s       r*   export_sub_broadcastzSub.export_sub_broadcast$   s    {$$:E % 
 
 IOOAq!$$++BJ77IOOA%%bj11EtQFQC6FGGGGGGr,   )r   N)__name__
__module____qualname__staticmethodr+   r/    r,   r*   r   r      s\        H H H \H, 
H 
H 
H \
H 
H 
Hr,   r   )numpyr    r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r4   r,   r*   <module>r8      s~   
      , , , , , , . . . . . .#H #H #H #H #H$ #H #H #H #H #Hr,   