§
    1�Åjx  ã                  ó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         ú\C:\Projects\Historical\venv\Lib\site-packages\onnx/backend/test/case/node/reducesumsquare.pyÚexport_do_not_keepdimsz&ReduceSumSquare.export_do_not_keepdims   s•  € à�	�	ˆÝŒx˜˜¥2¤8Ð,Ñ,Ô,ˆØˆåŒ{×$Ò$ØØ˜FÐ#Ø�KØð	 %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&�œ 4™œ­u°T©{¬{ÀXÐQRÂ]ÐSÑSÔSˆõ 	ØØ˜$�<Ø�IØAð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&�œ 4™œ­u°T©{¬{ÀXÐQRÂ]Ð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˜˜¥2¤8Ð,Ñ,Ô,ˆØˆåŒ{×$Ò$ØØ˜FÐ#Ø�KØð	 %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&�œ 4™œ­u°T©{¬{ÀXÐQRÂ]ÐSÑSÔSˆõ 	ØØ˜$�<Ø�IØ:ð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&�œ 4™œ­u°T©{¬{ÀXÐQRÂ]Ð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Ø�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&�œ 4™œ¨t¸hÈ!ºmÐLÑLÔLˆõ 	ØØ˜$�<Ø�IØGð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&�œ 4™œ¨t¸hÈ!º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˜˜¥B¤HÐ-Ñ-Ô-ˆØˆåŒ{×$Ò$ØØ˜FÐ#Ø�KØð	 %ñ 
ô 
ˆõ ŒxØ�!ˆf�q˜!�fÐ  A ¨¨A¨Ð/°1°b°'¸BÀ¸8Ð1DÐEÍRÌZð
ñ 
ô 
ˆõ ”&�œ 4™œ­u°T©{¬{ÀXÐQRÂ]ÐSÑSÔSˆõ 	ØØ˜$�<Ø�IØHð		
ñ 	
ô 	
ð 	
õ 	Œ	�Š�qÑÔÐÝŒy× Ò   b¨%Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝ”&�œ 4™œ­u°T©{¬{ÀXÐQRÂ]Ð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Ò5°eÑ<Ô<ˆÝŒx˜˜¥2¤8Ð,Ñ,Ô,ˆÝ”(�2œ8 M½¼Ð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<   