
    Si                    N    S SK Jr  S SKrS SKrS SKJr  S SKJr   " S S\5      r	g)    )annotationsN)Base)expectc                      \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS j5       r\SS	 j5       r\SS
 j5       rSrg)CumSum   c                 d   [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / SQ5      R                  [        R                  5      n[        XU/U/S	S
9  g )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       a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/cumsum.pyexport_cumsum_1dCumSum.export_cumsum_1d   s    {{$$XsFmcU$SHH./66rzzBxx{HH0188DtIs9IJ    c                 f   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r
   r   r      r   r   	exclusiver   r   )        r   r   r   r   test_cumsum_1d_exclusiver   r   r%   s       r'   export_cumsum_1d_exclusive!CumSum.export_cumsum_1d_exclusive   s    {{$$c6]SEQ % 
 HH./66rzzBxx{HH/077

CtIs9STr*   c                 f   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r
   r   r   r,   )r   r   reverser   r   )r         ,@      (@      "@r   test_cumsum_1d_reverser   r   r%   s       r'   export_cumsum_1d_reverseCumSum.export_cumsum_1d_reverse    s    {{$$c6]SE1 % 
 HH./66rzzBxx{HH1299"**EtIs9QRr*   c                 h   [         R                  R                  SSS/S/SSS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r
   r   r   r,   )r   r   r4   r.   r   r   )r5   r6   r7   r   r/    test_cumsum_1d_reverse_exclusiver   r   r%   s       r'   "export_cumsum_1d_reverse_exclusive)CumSum.export_cumsum_1d_reverse_exclusive*   s    {{$$c6]SE1PQ % 
 HH./66rzzBxx{HH0188DTQC6X	
r*   c                    [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r
   r   r   r   r   r   r   r   r   r         r   )r   r   r   r   g      @r7   test_cumsum_2d_axis_0r   
r   r   r   r    r!   r"   r#   reshaper$   r   r%   s       r'   export_cumsum_2d_axis_0CumSum.export_cumsum_2d_axis_06   s    {{$$=E % 

 HH34;;BJJGOOPVWxx{HH34;;BJJGOOPVWtIs9PQr*   c                    [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r
   r   r   r   r@   rA   r,   r   r   r   r   r7   r   test_cumsum_2d_axis_1r   rE   r%   s       r'   export_cumsum_2d_axis_1CumSum.export_cumsum_2d_axis_1B   s    {{$$=E % 

 HH34;;BJJGOOPVWxx{HH45<<RZZHPPQWXtIs9PQr*   c                    [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r
   r   r   r   r@   rA   rJ   test_cumsum_2d_negative_axisr   rE   r%   s       r'   export_cumsum_2d_negative_axis%CumSum.export_cumsum_2d_negative_axisN   s    {{$$=E % 

 HH34;;BJJGOOPVWxx|HH45<<RZZHPPQWXtIs9WXr*   c                    [         R                  R                  SSS/S/S9n [        R                  " / SQ5      R                  [        R                  5      R                  S5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      R                  S5      n[        XU/U/S
S9  g )Nr   r
   r   r   r   )r,   rB   rC            rA   r   )r,   rB   rC   rU      	   test_cumsum_2d_int32r   )	r   r   r   r    r!   r"   r$   rF   r   r%   s       r'   export_cumsum_2d_int32CumSum.export_cumsum_2d_int32Z   s    {{$$=E % 

 HH'(//9AA&Ixx{HH'(//9AA&ItIs9OPr*   c                 f   [         R                  R                  SSS/S/SS9n [        R                  " / SQ5      R                  [        R                  5      n[        R                  " S5      n[        R                  " / S	Q5      R                  [        R                  5      n[        XU/U/S
S9  g )Nr   r
   r   r   r,   r-   )r,   rB   rC   rT   rU   r   )r   r,   rC   rV   
   test_cumsum_1d_int32_exclusiver   )r   r   r   r    r!   r"   r$   r   r%   s       r'    export_cumsum_1d_int32_exclusive'CumSum.export_cumsum_1d_int32_exclusivef   s    {{$$c6]SEQ % 
 HH_%,,RXX6xx{HH%&--bhh7TQC6V	
r*    N)returnNone)__name__
__module____qualname____firstlineno__staticmethodr(   r1   r9   r=   rG   rL   rQ   rZ   r_   __static_attributes__ra   r*   r'   r   r      s    K K U U S S 	
 	
 	R 	R 	R 	R 	Y 	Y 	Q 	Q 	
 	
r*   r   )

__future__r   numpyr    r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   ra   r*   r'   <module>rn      s$    #   , .c
T c
r*   