
    /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dS )
ReduceSumSquarereturnNonec                    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         p/home/longshao/multi-rider-rag/.venv/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(   r7   s         r:   export_keepdimszReduceSumSquare.export_keepdims6   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/   r0   r1   r   r3   r4   r5   r6   r7   s         r:   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(   r7   s         r:   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   )r8   r   reduced_shaper9   r   r   r   s          r:   export_empty_setz ReduceSumSquare.export_empty_set   s    		!		{$$F#K	 % 
 
 x"*---55e<<x28,,,(28MDDDEE$<I3		
 	
 	
 	
 	
 	
r<   N)r   r	   )	__name__
__module____qualname__staticmethodr;   r@   rD   rI   rO    r<   r:   r   r      s        %
 %
 %
 \%
N %
 %
 %
 \%
N #
 #
 #
 \#
J %
 %
 %
 \%
N 
 
 
 \
 
 
r<   r   )

__future__r   numpyr)   r,   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rT   r<   r:   <module>rY      s    # " " " " "      , , , , , , . . . . . .u
 u
 u
 u
 u
d u
 u
 u
 u
 u
r<   