
    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
dS )	ReduceSum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          |	                                          |dk              }t          |||g|gd           t           j                            d           t           j                            dd|                               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_do_not_keepdims_exampler   r   name&test_reduce_sum_do_not_keepdims_randomnparrayint64onnxhelper	make_nodefloat32sumtupletolistr   randomseeduniformastypeshaper   r   noder   r   s         a/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/reducesum.pyexport_do_not_keepdimsz ReduceSum.export_do_not_keepdims   s   		x28,,,{$$ 09+PX % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &E$++--$8$88q=QQQ 	$<I:		
 	
 	
 	
 		qy  b%0077
CC&E$++--$8$88q=QQQ$<I9		
 	
 	
 	
 	
 	
    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          |	                                          |dk              }t          |||g|gd           t           j                            d           t           j                            dd|                               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_keepdims_exampler"   r   r$   test_reduce_sum_keepdims_randomr&   r5   s         r8   export_keepdimszReduceSum.export_keepdims2   s   		x28,,,{$$ 09+PX % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &E$++--$8$88q=QQQ 	$<I3		
 	
 	
 	
 		qy  b%0077
CC&E$++--$8$88q=QQQ$<I2		
 	
 	
 	
 	
 	
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        |d |dk              }t          |||g|gd           t           j	        
                    d           t           j	                            dd|                               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   -test_reduce_sum_default_axes_keepdims_exampler"   r   r$   ,test_reduce_sum_default_axes_keepdims_random)r'   r(   r)   r*   r+   r,   r-   r.   r   r1   r2   r3   r4   r5   s         r8   export_default_axes_keepdimsz&ReduceSum.export_default_axes_keepdimsW   ss   		x"(+++{$$ 09+PX % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &D8q=AAA 	$<I@		
 	
 	
 	
 		qy  b%0077
CC&D8q=AAA$<I?		
 	
 	
 	
 	
 	
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          |	                                          |dk              }t          |||g|gd           t           j                            d           t           j                            dd|                               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_negative_axes_keepdims_exampler"   r   r$   -test_reduce_sum_negative_axes_keepdims_randomr&   r5   s         r8   export_negative_axes_keepdimsz'ReduceSum.export_negative_axes_keepdimsz   s   		xBH---{$$ 09+PX % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 &E$++--$8$88q=QQQ 	$<IA		
 	
 	
 	
 		qy  b%0077
CC&E$++--$8$88q=QQQ$<I@		
 	
 	
 	
 	
 	
r:   c                  l   g d} d}t           j                            dddgdg|d          }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        g t          j                  }t          j        |          }t          |||g|gd           t          j        	                    d           t          j        
                    dd|                               t          j                  }t          j        |          }t          |||g|gd           d S )Nr	   r   r   r   r   r   T)r   r   r   noop_with_empty_axesr   r
   r   r   r   r   r   r   r   r   r   r   -test_reduce_sum_empty_axes_input_noop_exampler"   r   r$   rF   )r*   r+   r,   r'   r(   r-   r)   r   r1   r2   r3   r4   )r6   r   r7   r   r   r   s         r8   export_empty_axes_input_noopz&ReduceSum.export_empty_axes_input_noop   s^   		{$$F#K!% % 
 
 x!fq!fAA/1b'B81DERZ
 
 
 x"(+++(4.. 	$<I@		
 	
 	
 	
 		qy  b%0077
CC(4..$<I@		
 	
 	
 	
 	
 	
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 )z-Test case with the reduced-axis of size zero.r   r   r   r   )r   r   r   r   r   r   r   r   r   test_reduce_sum_empty_setr"   N)
r*   r+   r,   r'   r(   r-   reshaper)   zerosr   r6   r   reduced_shaper7   r   r   r   s          r8   export_empty_setzReduceSum.export_empty_set   s     		!		{$$F#K	 % 
 
 x"*---55e<<x28,,,(28MDDDEE$<I,		
 	
 	
 	
 	
 	
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        g t          j        	                              |          }t          |||g|gd           dS )z1Test case with the non-reduced-axis of size zero.rM   r   )r   r   r   r   r   r   r   r   r   r   /test_reduce_sum_empty_set_non_reduced_axis_zeror"   N)	r*   r+   r,   r'   r(   r-   rO   r)   r   rQ   s          r8   export_non_reduced_axis_zeroz&ReduceSum.export_non_reduced_axis_zero   s     		!		{$$F#K	 % 
 
 x"*---55e<<x28,,,(2RZ00088GG$<IB		
 	
 	
 	
 	
 	
r:   )r   N)__name__
__module____qualname__staticmethodr9   r>   rB   rG   rK   rS   rV    r:   r8   r   r      s        "
 "
 "
 \"
H "
 "
 "
 \"
H  
  
  
 \ 
D "
 "
 "
 \"
H $
 $
 $
 \$
L 
 
 
 \
0 
 
 
 \
 
 
r:   r   )numpyr'   r*   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r[   r:   r8   <module>r_      s~   
      , , , , , , . . . . . .j
 j
 j
 j
 j
 j
 j
 j
 j
 j
r:   