
    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dS )	ReduceSumSquarereturnNc                     g d} t          j        dgt           j                  }d}t          j                            dddgdg|	          }t          j        dd
gddggddgddggddgddgggt           j                  }t          j        t          j        |          t          |          |dk              }t          |||g|gd           t           j                            d           t           j                            dd|                               t           j                  }t          j        t          j        |          t          |          |dk              }t          |||g|gd           d S )N      r      dtyper   r   dataaxesreducedinputsoutputskeepdimsr   r
                  	   
         axisr   .test_reduce_sum_square_do_not_keepdims_exampler   r   name-test_reduce_sum_square_do_not_keepdims_randomnparrayint64onnxhelper	make_nodefloat32sumsquaretupler   randomseeduniformastypeshaper   r   noder   r   s         g/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/reducesumsquare.pyexport_do_not_keepdimsz&ReduceSumSquare.export_do_not_keepdims   s   		x28,,,{$$F#K	 % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &4uT{{XQR]SSS 	$<IA		
 	
 	
 	
 		qy  b%0077
CC&4uT{{XQR]SSS$<I@		
 	
 	
 	
 	
 	
    c                     g d} t          j        dgt           j                  }d}t          j                            dddgdg|          }t          j        dd	gd
dggddgddggddgddgggt           j                  }t          j        t          j        |          t          |          |dk              }t          |||g|gd           t           j                            d           t           j                            dd|                               t           j                  }t          j        t          j        |          t          |          |dk              }t          |||g|gd           d S )Nr	   r   r   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   'test_reduce_sum_square_keepdims_exampler"   r   r$   &test_reduce_sum_square_keepdims_randomr&   r5   s         r8   export_keepdimszReduceSumSquare.export_keepdims5   s   		x28,,,{$$F#K	 % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &4uT{{XQR]SSS 	$<I:		
 	
 	
 	
 		qy  b%0077
CC&4uT{{XQR]SSS$<I9		
 	
 	
 	
 	
 	
r:   c                     g d} t          j        g t           j                  }d}t          j                            dddgdg|          }t          j        dd	gd
dggddgddggddgddgggt           j                  }t          j        t          j        |          d |dk              }t          |||g|gd           t           j
                            d           t           j
                            dd|                               t           j                  }t          j        t          j        |          d |dk              }t          |||g|gd           d S )Nr	   r   r   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   4test_reduce_sum_square_default_axes_keepdims_exampler"   r   r$   3test_reduce_sum_square_default_axes_keepdims_random)r'   r(   r)   r*   r+   r,   r-   r.   r/   r   r1   r2   r3   r4   r5   s         r8   export_default_axes_keepdimsz,ReduceSumSquare.export_default_axes_keepdims]   s   		x"(+++{$$F#K	 % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &4th!mLLL 	$<IG		
 	
 	
 	
 		qy  b%0077
CC&4th!mLLL$<IF		
 	
 	
 	
 	
 	
r:   c                     g d} t          j        dgt           j                  }d}t          j                            dddgdg|	          }t          j        dd
gddggddgddggddgddgggt           j                  }t          j        t          j        |          t          |          |dk              }t          |||g|gd           t           j                            d           t           j                            dd|                               t           j                  }t          j        t          j        |          t          |          |dk              }t          |||g|gd           d S )Nr	   r   r   r   r   r   r   r   r   r
   r   r   r   r   r   r   r   r   r   r   5test_reduce_sum_square_negative_axes_keepdims_exampler"   r   r$   4test_reduce_sum_square_negative_axes_keepdims_randomr&   r5   s         r8   export_negative_axes_keepdimsz-ReduceSumSquare.export_negative_axes_keepdims   s   		xBH---{$$F#K	 % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &4uT{{XQR]SSS 	$<IH		
 	
 	
 	
 		qy  b%0077
CC&4uT{{XQR]SSS$<IG		
 	
 	
 	
 	
 	
r:   c                     g d} d}g d}t           j                            dddgdg|          }t          j        g t          j        	                              |           }t          j        dgt          j        	          }t          j        t          j        |t          j        	                    }t          |||g|gd
           d S )N)r   r   r   r   )r   r   r   r   r   r   r   r   r    test_reduce_sum_square_empty_setr"   )
r*   r+   r,   r'   r(   r-   reshaper)   zerosr   )r6   r   reduced_shaper7   r   r   r   s          r8   export_empty_setz ReduceSumSquare.export_empty_set   s    		!		{$$F#K	 % 
 
 x"*---55e<<x28,,,(28MDDDEE$<I3		
 	
 	
 	
 	
 	
r:   )r   N)	__name__
__module____qualname__staticmethodr9   r>   rB   rG   rM    r:   r8   r   r      s        %
 %
 %
 \%
N %
 %
 %
 \%
N #
 #
 #
 \#
J %
 %
 %
 \%
N 
 
 
 \
 
 
r:   r   )numpyr'   r*   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rR   r:   r8   <module>rV      s~   
      , , , , , , . . . . . .u
 u
 u
 u
 u
d u
 u
 u
 u
 u
r:   