§
    1�ÅjO  ã                  ó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
edd
„¦   «         Zedd„¦   «         ZdS )ÚCumSumÚreturnÚNonec                 óp  — t           j                             dddgdg¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «        }t          j        d¦  «        }t          j        g d¢¦  «                             t          j        ¦  «        }t          | ||g|gd	¬
¦  «         d S )Nr   ÚxÚaxisÚy©ÚinputsÚoutputs©ç      ð?ç       @ç      @ç      @ç      @r   )r   r   ç      @ç      $@ç      .@Útest_cumsum_1d©r   r   Úname©	ÚonnxÚhelperÚ	make_nodeÚnpÚarrayÚastypeÚfloat64Úint32r   ©Únoder   r   r   s       úSC:\Projects\Historical\venv\Lib\site-packages\onnx/backend/test/case/node/cumsum.pyÚexport_cumsum_1dzCumSum.export_cumsum_1d   s¡   € åŒ{×$Ò$ X°s¸F°mÈcÈUÐ$ÑSÔSˆÝŒHÐ.Ð.Ð.Ñ/Ô/×6Ò6µr´zÑBÔBˆÝŒx˜‰{Œ{ˆÝŒHÐ0Ð0Ð0Ñ1Ô1×8Ò8½¼ÑDÔDˆÝˆt˜Q ˜I°¨sÐ9IÐJÑJÔJÐJÐJÐJó    c                 ór  — t           j                             dddgdgd¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   é   ©r   r   Ú	exclusiver   r   )ç        r   r   r   r   Útest_cumsum_1d_exclusiver   r   r&   s       r(   Úexport_cumsum_1d_exclusivez!CumSum.export_cumsum_1d_exclusive   s¬   € åŒ{×$Ò$Ø˜c 6˜]°S°EÀQð %ñ 
ô 
ˆõ ŒHÐ.Ð.Ð.Ñ/Ô/×6Ò6µr´zÑBÔBˆÝŒx˜‰{Œ{ˆÝŒHÐ/Ð/Ð/Ñ0Ô0×7Ò7½¼
ÑCÔCˆÝˆt˜Q ˜I°¨sÐ9SÐTÑTÔTÐTÐTÐTr*   c                 ór  — t           j                             dddgdgd¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r,   )r   r   Úreverser   r   )r   ç      ,@ç      (@ç      "@r   Útest_cumsum_1d_reverser   r   r&   s       r(   Úexport_cumsum_1d_reversezCumSum.export_cumsum_1d_reverse    s¬   € åŒ{×$Ò$Ø˜c 6˜]°S°EÀ1ð %ñ 
ô 
ˆõ ŒHÐ.Ð.Ð.Ñ/Ô/×6Ò6µr´zÑBÔBˆÝŒx˜‰{Œ{ˆÝŒHÐ1Ð1Ð1Ñ2Ô2×9Ò9½"¼*ÑEÔEˆÝˆt˜Q ˜I°¨sÐ9QÐRÑRÔRÐRÐRÐRr*   c                 ót  — t           j                             dddgdgdd¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r,   )r   r   r3   r.   r   r   )r4   r5   r6   r   r/   Ú test_cumsum_1d_reverse_exclusiver   r   r&   s       r(   Ú"export_cumsum_1d_reverse_exclusivez)CumSum.export_cumsum_1d_reverse_exclusive*   s¼   € åŒ{×$Ò$Ø˜c 6˜]°S°EÀ1ÐPQð %ñ 
ô 
ˆõ ŒHÐ.Ð.Ð.Ñ/Ô/×6Ò6µr´zÑBÔBˆÝŒx˜‰{Œ{ˆÝŒHÐ0Ð0Ð0Ñ1Ô1×8Ò8½¼ÑDÔDˆÝØ˜!˜T˜¨Q¨CÐ6Xð	
ñ 	
ô 	
ð 	
ð 	
ð 	
r*   c                 ó¼  — t           j                             dddgdg¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «                             d¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «                             d¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r   ©r   r   r   r   r   r   ©é   é   r   )r   r   r   r   g      @r6   Útest_cumsum_2d_axis_0r   ©
r   r   r    r!   r"   r#   r$   Úreshaper%   r   r&   s       r(   Úexport_cumsum_2d_axis_0zCumSum.export_cumsum_2d_axis_06   sÊ   € åŒ{×$Ò$ØØ˜�=Ø�Eð %ñ 
ô 
ˆõ
 ŒHÐ3Ð3Ð3Ñ4Ô4×;Ò;½B¼JÑGÔG×OÒOÐPVÑWÔWˆÝŒx˜‰{Œ{ˆÝŒHÐ3Ð3Ð3Ñ4Ô4×;Ò;½B¼JÑGÔG×OÒOÐPVÑWÔWˆÝˆt˜Q ˜I°¨sÐ9PÐQÑQÔQÐQÐQÐQr*   c                 ó¼  — t           j                             dddgdg¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «                             d¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «                             d¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r   r=   r>   r,   ©r   r   r   r   r6   r   Útest_cumsum_2d_axis_1r   rB   r&   s       r(   Úexport_cumsum_2d_axis_1zCumSum.export_cumsum_2d_axis_1B   sÊ   € åŒ{×$Ò$ØØ˜�=Ø�Eð %ñ 
ô 
ˆõ
 ŒHÐ3Ð3Ð3Ñ4Ô4×;Ò;½B¼JÑGÔG×OÒOÐPVÑWÔWˆÝŒx˜‰{Œ{ˆÝŒHÐ4Ð4Ð4Ñ5Ô5×<Ò<½R¼ZÑHÔH×PÒPÐQWÑXÔXˆÝˆt˜Q ˜I°¨sÐ9PÐQÑQÔQÐQÐQÐQr*   c                 ó¼  — t           j                             dddgdg¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «                             d¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «                             d¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r   r=   r>   éÿÿÿÿrF   Útest_cumsum_2d_negative_axisr   rB   r&   s       r(   Úexport_cumsum_2d_negative_axisz%CumSum.export_cumsum_2d_negative_axisN   sÊ   € åŒ{×$Ò$ØØ˜�=Ø�Eð %ñ 
ô 
ˆõ
 ŒHÐ3Ð3Ð3Ñ4Ô4×;Ò;½B¼JÑGÔG×OÒOÐPVÑWÔWˆÝŒx˜‰|Œ|ˆÝŒHÐ4Ð4Ð4Ñ5Ô5×<Ò<½R¼ZÑHÔH×PÒPÐQWÑXÔXˆÝˆt˜Q ˜I°¨sÐ9WÐXÑXÔXÐXÐXÐXr*   c                 ó¼  — t           j                             dddgdg¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «                             d¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «                             d¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r   )r,   r?   r@   é   é   é   r>   r   )r,   r?   r@   rO   é   é	   Útest_cumsum_2d_int32r   )	r   r   r    r!   r"   r#   r%   rC   r   r&   s       r(   Úexport_cumsum_2d_int32zCumSum.export_cumsum_2d_int32Z   sÈ   € åŒ{×$Ò$ØØ˜�=Ø�Eð %ñ 
ô 
ˆõ
 ŒHÐ'Ð'Ð'Ñ(Ô(×/Ò/µ´Ñ9Ô9×AÒAÀ&ÑIÔIˆÝŒx˜‰{Œ{ˆÝŒHÐ'Ð'Ð'Ñ(Ô(×/Ò/µ´Ñ9Ô9×AÒAÀ&ÑIÔIˆÝˆt˜Q ˜I°¨sÐ9OÐPÑPÔPÐPÐPÐPr*   c                 ór  — t           j                             dddgdgd¬¦  «        } t          j        g d¢¦  «                             t          j        ¦  «        }t          j        d¦  «        }t          j        g d	¢¦  «                             t          j        ¦  «        }t          | ||g|gd
¬¦  «         d S )Nr   r   r   r   r,   r-   )r,   r?   r@   rN   rO   r   )r   r,   r@   rP   é
   Útest_cumsum_1d_int32_exclusiver   )r   r   r    r!   r"   r#   r%   r   r&   s       r(   Ú export_cumsum_1d_int32_exclusivez'CumSum.export_cumsum_1d_int32_exclusivef   s¶   € åŒ{×$Ò$Ø˜c 6˜]°S°EÀQð %ñ 
ô 
ˆõ ŒH�_�_�_Ñ%Ô%×,Ò,­R¬XÑ6Ô6ˆÝŒx˜‰{Œ{ˆÝŒHÐ%Ð%Ð%Ñ&Ô&×-Ò-­b¬hÑ7Ô7ˆÝØ˜!˜T˜¨Q¨CÐ6Vð	
ñ 	
ô 	
ð 	
ð 	
ð 	
r*   N)r   r	   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr)   r1   r8   r;   rD   rH   rL   rT   rX   © r*   r(   r   r      sN  € € € € € ØðKð Kð Kñ „\ðKð ðUð Uð Uñ „\ðUð ðSð Sð Sñ „\ðSð ð	
ð 	
ð 	
ñ „\ð	
ð ð	Rð 	Rð 	Rñ „\ð	Rð ð	Rð 	Rð 	Rñ „\ð	Rð ð	Yð 	Yð 	Yñ „\ð	Yð ð	Qð 	Qð 	Qñ „\ð	Qð ð	
ð 	
ð 	
ñ „\ð	
ð 	
ð 	
r*   r   )
Ú
__future__r   Únumpyr!   r   Úonnx.backend.test.case.baser   Úonnx.backend.test.case.noder   r   r]   r*   r(   ú<module>rb      s�   ðð #Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð à €€€Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø .Ð .Ð .Ð .Ð .Ð .ðc
ð c
ð c
ð c
ð c
ˆTñ c
ô c
ð c
ð c
ð c
r*   