
    /jj                    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 )CumProdreturnNonec                    t           j                            dddgdg          } t          j        g d                              t          j                  }t          j        dt          j                  }t          j        g d	                              t          j                  }t          | ||g|gd
           d S )Nr   xaxisyinputsoutputs      ?       @      @      @      @r   dtype)r   r         @      8@      ^@test_cumprod_1dr   r   name	onnxhelper	make_nodenparrayastypefloat64int32r   noder   r   r   s       h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/cumprod.pyexport_cumprod_1dzCumProd.export_cumprod_1d   s    {$$YV}se$TTH...//66rzBBx***H1112299"*EEtQIs9JKKKKKK    c                    t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        dt          j        	          }t          j        g d
                              t          j                  }t          | ||g|gd           d S )Nr   r   r   r      r   r   	exclusiver   r   r   )r   r   r   r   r   test_cumprod_1d_exclusiver   r   r(   s       r*   export_cumprod_1d_exclusivez#CumProd.export_cumprod_1d_exclusive   s    {$$sFmcUa % 
 
 H...//66rzBBx***H///0077
CCtQIs9TUUUUUUr,   c                    t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        dt          j        	          }t          j        g d
                              t          j                  }t          | ||g|gd           d S )Nr   r   r   r   r.   )r   r   reverser   r   r   )r   r         N@      4@r   test_cumprod_1d_reverser   r   r(   s       r*   export_cumprod_1d_reversez!CumProd.export_cumprod_1d_reverse    s    {$$sFmcUA % 
 
 H...//66rzBBx***H44455<<RZHHtQIs9RSSSSSSr,   c                    t           j                            dddgdgdd          } t          j        g d                              t          j                  }t          j        dt          j        	          }t          j        g d
                              t          j                  }t          | ||g|gd           d S )Nr   r   r   r   r.   )r   r   r4   r0   r   r   r   )r   r5   r6   r   r   !test_cumprod_1d_reverse_exclusiver   r   r(   s       r*   #export_cumprod_1d_reverse_exclusivez+CumProd.export_cumprod_1d_reverse_exclusive*   s    {$$sFmcUAQR % 
 
 H...//66rzBBx***H22233::2:FFt9C4		
 	
 	
 	
 	
 	
r,   c                    t           j                            dddgdg          } t          j        g d                              t          j                                      d          }t          j        dt          j        	          }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   r   g      $@g      2@test_cumprod_2d_axis_0r   
r    r!   r"   r#   r$   r%   r&   reshaper'   r   r(   s       r*   export_cumprod_2d_axis_0z CumProd.export_cumprod_2d_axis_09   s    {$$=E % 
 

 H33344;;BJGGOOPVWWx***H55566VBJWV__ 	

 	tQIs9QRRRRRRr,   c                    t           j                            dddgdg          } t          j        g d                              t          j                                      d          }t          j        dt          j        	          }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   r6   r   test_cumprod_2d_axis_1r   rB   r(   s       r*   export_cumprod_2d_axis_1z CumProd.export_cumprod_2d_axis_1I   s    {$$=E % 
 

 H33344;;BJGGOOPVWWx***H66677VBJWV__ 	

 	tQIs9QRRRRRRr,   c                    t           j                            dddgdg          } t          j        g d                              t          j                                      d          }t          j        dt          j        	          }t          j        g d
                              t          j                                      d          }t          | ||g|gd           d S )Nr   r   r   r   r   r=   r>   r   rF   test_cumprod_2d_negative_axisr   rB   r(   s       r*   export_cumprod_2d_negative_axisz'CumProd.export_cumprod_2d_negative_axisY   s    {$$=E % 
 

 H33344;;BJGGOOPVWWx"(+++H66677VBJWV__ 	

 	!TQC6U	
 	
 	
 	
 	
 	
r,   c                    t           j                            dddgdg          } t          j        g d                              t          j                                      d          }t          j        dt          j        	          }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@   rN   
      test_cumprod_2d_int32r   )	r    r!   r"   r#   r$   r%   r'   rC   r   r(   s       r*   export_cumprod_2d_int32zCumProd.export_cumprod_2d_int32k   s    {$$=E % 
 

 H'''((//99AA&IIx***H)))**11"(;;CCFKKtQIs9PQQQQQQr,   c                    t           j                            dddgdgd          } t          j        g d                              t          j                  }t          j        dt          j        	          }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.   r?   rP      test_cumprod_1d_int32_exclusiver   )r    r!   r"   r#   r$   r%   r'   r   r(   s       r*   !export_cumprod_1d_int32_exclusivez)CumProd.export_cumprod_1d_int32_exclusivew   s    {$$sFmcUa % 
 
 H___%%,,RX66x***H%%%&&--bh77!TQC6W	
 	
 	
 	
 	
 	
r,   N)r   r	   )__name__
__module____qualname__staticmethodr+   r2   r8   r;   rD   rH   rL   rT   rX    r,   r*   r   r      sJ       L L L \L V V V \V T T T \T 
 
 
 \
 S S S \S S S S \S 
 
 
 \
" 	R 	R 	R \	R 	
 	
 	
 \	
 	
 	
r,   r   )

__future__r   numpyr#   r    onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r]   r,   r*   <module>rb      s    # " " " " "      , , , , , , . . . . . .t
 t
 t
 t
 t
d t
 t
 t
 t
 t
r,   