§
    1�Å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edd„¦   «         Z	edd	„¦   «         Z
d
S )Ú	ReduceSumÚ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          | 	                    ¦   «         ¦  «        |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         úVC:\Projects\Historical\venv\Lib\site-packages\onnx/backend/test/case/node/reducesum.pyÚexport_do_not_keepdimsz ReduceSum.export_do_not_keepdims   s•  € à�	�	ˆÝŒx˜˜¥2¤8Ð,Ñ,Ô,ˆØˆåŒ{×$Ò$Ø ¨Ð 0¸9¸+ÐPXð %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&˜¥E¨$¯+ª+©-¬-Ñ$8Ô$8À8ÈqÂ=ÐQÑQÔQˆõ 	ØØ˜$�<Ø�IØ:ð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&˜¥E¨$¯+ª+©-¬-Ñ$8Ô$8À8ÈqÂ=Ð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(   r7   s         r:   Úexport_keepdimszReduceSum.export_keepdims3   s•  € à�	�	ˆÝŒx˜˜¥2¤8Ð,Ñ,Ô,ˆØˆåŒ{×$Ò$Ø ¨Ð 0¸9¸+ÐPXð %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&˜¥E¨$¯+ª+©-¬-Ñ$8Ô$8À8ÈqÂ=ÐQÑQÔQˆõ 	ØØ˜$�<Ø�IØ3ð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&˜¥E¨$¯+ª+©-¬-Ñ$8Ô$8À8ÈqÂ=Ð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/   r0   r   r3   r4   r5   r6   r7   s         r:   Úexport_default_axes_keepdimsz&ReduceSum.export_default_axes_keepdimsX   ss  € à�	�	ˆÝŒx˜¥"¤(Ð+Ñ+Ô+ˆØˆåŒ{×$Ò$Ø ¨Ð 0¸9¸+ÐPXð %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&˜ D°8¸q²=ÐAÑAÔAˆõ 	ØØ˜$�<Ø�IØ@ð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&˜ D°8¸q²=Ð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(   r7   s         r:   Úexport_negative_axes_keepdimsz'ReduceSum.export_negative_axes_keepdims{   s•  € à�	�	ˆÝŒx˜˜¥B¤HÐ-Ñ-Ô-ˆØˆåŒ{×$Ò$Ø ¨Ð 0¸9¸+ÐPXð %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&˜¥E¨$¯+ª+©-¬-Ñ$8Ô$8À8ÈqÂ=ÐQÑQÔQˆõ 	ØØ˜$�<Ø�IØAð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&˜¥E¨$¯+ª+©-¬-Ñ$8Ô$8À8ÈqÂ=Ð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&   Ú%test_reduce_sum_empty_axes_input_noop)r,   r-   r.   r)   r*   r/   r+   r   r3   r4   r5   r6   )r8   r   r9   r   r   r   s         r:   Úexport_empty_axes_input_noopz&ReduceSum.export_empty_axes_input_noop    s^  € à�	�	ˆØˆåŒ{×$Ò$ØØ˜FÐ#Ø�KØØ!%ð %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ Œx˜¥"¤(Ð+Ñ+Ô+ˆÝ”(˜4‘.”.ˆõ 	ØØ˜$�<Ø�IØ@ð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”(˜4‘.”.ˆåØØ˜$�<Ø�IØ8ð		
ñ 	
ô 	
ð 	
ð 	
ð 	
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   ©r8   r   Úreduced_shaper9   r   r   r   s          r:   Úexport_empty_setzReduceSum.export_empty_setÇ   sÛ   € ð �	�	ˆØˆØ!˜	˜	ˆåŒ{×$Ò$ØØ˜FÐ#Ø�KØð	 %ñ 
ô 
ˆõ Œx˜¥"¤*Ð-Ñ-Ô-×5Ò5°eÑ<Ô<ˆÝŒx˜˜¥2¤8Ð,Ñ,Ô,ˆÝ”(�2œ8 M½¼Ð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.rP   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/   rR   r+   r   rT   s          r:   Úexport_non_reduced_axis_zeroz&ReduceSum.export_non_reduced_axis_zeroà   sß   € ð �	�	ˆØˆØ!˜	˜	ˆåŒ{×$Ò$ØØ˜FÐ#Ø�KØð	 %ñ 
ô 
ˆõ Œx˜¥"¤*Ð-Ñ-Ô-×5Ò5°eÑ<Ô<ˆÝŒx˜˜¥2¤8Ð,Ñ,Ô,ˆÝ”(˜2¥R¤ZÐ0Ñ0Ô0×8Ò8¸ÑGÔGˆåØØ˜$�<Ø�IØBð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r<   N)r   r	   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr;   r@   rD   rI   rN   rV   rY   © r<   r:   r   r      s÷   € € € € € Øð"
ð "
ð "
ñ „\ð"
ðH ð"
ð "
ð "
ñ „\ð"
ðH ð 
ð  
ð  
ñ „\ð 
ðD ð"
ð "
ð "
ñ „\ð"
ðH ð$
ð $
ð $
ñ „\ð$
ðL ð
ð 
ð 
ñ „\ð
ð0 ð
ð 
ð 
ñ „\ð
ð 
ð 
r<   r   )
Ú
__future__r   Únumpyr)   r,   Úonnx.backend.test.case.baser   Úonnx.backend.test.case.noder   r   r^   r<   r:   ú<module>rc      s�   ðð #Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð à €€€Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø .Ð .Ð .Ð .Ð .Ð .ðj
ð j
ð j
ð j
ð j
�ñ j
ô j
ð j
ð j
ð j
r<   