
    /jE.                    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edd            ZdS )SplitreturnNonec            
     :   t          j        g d                              t           j                  } t          j                            ddgg dd          }t          j        ddg                              t           j                  t          j        d	d
g                              t           j                  t          j        ddg                              t           j                  g}t          || g|dt          j                            dd          g           t          j        ddg                              t           j	                  }t          j                            dddgddgd          }t          j        ddg                              t           j                  t          j        g d                              t           j                  g}t          || |g|dt          j                            dd          g           d S )N      ?       @      @      @      @      @r   inputoutput_1output_2output_3r   inputsoutputsaxisr   r   r   r   r   r   !test_split_equal_parts_1d_opset13    r   r   nameopset_imports      splitr   r   r   r   r   r   $test_split_variable_parts_1d_opset13
nparrayastypefloat32onnxhelper	make_noder   make_opsetidint64
node_inputnodeexpected_outputsr#   s       f/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/split.pyexport_1d_opset13zSplit.export_1d_opset13   s   X<<<==DDRZPP
{$$9888	 % 
 
 Hc3Z  ''
33Hc3Z  ''
33Hc3Z  ''
33

 	<$4;33B;;<	
 	
 	
 	
 !Q  ''11{$$W%,	 % 
 
 Hc3Z  ''
33H)))**11"*==
 	&$7;33B;;<	
 	
 	
 	
 	
 	
    c            
        t          j        g dg dg                              t           j                  } t          j                            ddgddgd          }t          j        g d	g d
g                              t           j                  t          j        g dg dg                              t           j                  g}t          || g|dt          j                            dd          g           t          j        ddg                              t           j	                  }t          j                            dddgddgd          }t          j        ddgddgg                              t           j                  t          j        g dg dg                              t           j                  g}t          || |g|dt          j                            dd          g           d S )Nr         @       @      "@      $@      &@      (@r   r   r   r      r   r   r   r   r9   r:   r;   r   r   r   r<   r=   r>   !test_split_equal_parts_2d_opset13r   r   r   r!   r"   r#   r   r   r9   r:   r$   r;   r<   r=   r>   $test_split_variable_parts_2d_opset13r&   r0   s       r4   export_2d_opset13zSplit.export_2d_opset13:   s   X+++-N-N-NO
 

&

 	 {$$WI
J/Ga % 
 

 Hooo788??
KKHooo'9'9'9:;;BB2:NN

 	<$4;33B;;<	
 	
 	
 	
 !Q  ''11{$$W%,	 % 
 
 HsCj3*-..55bjAAH***,C,C,CDEELL
 
 	&$7;33B;;<	
 	
 	
 	
 	
 	
r6   c            
     6   t          j        g d                              t           j                  } t          j                            ddgg d          }t          j        ddg                              t           j                  t          j        dd	g                              t           j                  t          j        d
dg                              t           j                  g}t          || g|dt          j                            dd          g           t          j        ddg                              t           j	                  }t          j                            dddgddg          }t          j        ddg                              t           j                  t          j        g d                              t           j                  g}t          || |g|dt          j                            dd          g           d S )Nr   r   r   r   r   r   r   r   r   r   r   r   +test_split_equal_parts_default_axis_opset13r   r   r   r!   r"   r#   r   r   r$   .test_split_variable_parts_default_axis_opset13r&   r0   s       r4   export_default_values_opset13z#Split.export_default_values_opset13h   s   X<<<==DDRZPP
 {$$WI/S/S/S % 
 

 Hc3Z  ''
33Hc3Z  ''
33Hc3Z  ''
33

 	<$>;33B;;<	
 	
 	
 	
 !Q  ''11{$$Wg.Z8P % 
 

 Hc3Z  ''
33H)))**11"*==
 	&$A;33B;;<	
 	
 	
 	
 	
 	
r6   c            
        t          j        g                               t           j                  } t          j        g d                              t           j                  }t
          j                            dddgg d          }t          j        g                               t           j                  t          j        g                               t           j                  t          j        g                               t           j                  g}t          || |g|dt
          j        	                    dd	          g
           d S )Nr   r   r   r   r   r#   r   rI   #test_split_zero_size_splits_opset13r   r   r   )
r'   r(   r)   r*   r/   r+   r,   r-   r   r.   r1   r#   r2   r3   s       r4   export_zero_size_splits_opset13z%Split.export_zero_size_splits_opset13   s    Xb\\((44
 ##**2844{$$W%888 % 
 
 HRLL
++HRLL
++HRLL
++

 	&$6;33B;;<	
 	
 	
 	
 	
 	
r6   c                    t          j        g d                              t           j                  } t          j                            ddgg ddd          }t          j        dd	g                              t           j                  t          j        d
dg                              t           j                  t          j        ddg                              t           j                  g}t          || g|d           t          j        ddg                              t           j                  }t          j                            dddgddgd          }t          j        dd	g                              t           j                  t          j        g d                              t           j                  g}t          || |g|d           d S )Nr   r   r   r   r      r   r   r   num_outputsr   r   r   r   r   r   !test_split_equal_parts_1d_opset18r   r   r   r!   r"   r#   r   r   r   r$   $test_split_variable_parts_1d_opset18	r'   r(   r)   r*   r+   r,   r-   r   r/   r0   s       r4   export_1d_opset18zSplit.export_1d_opset18   s   X<<<==DDRZPP
{$$9888 % 
 
 Hc3Z  ''
33Hc3Z  ''
33Hc3Z  ''
33

 	<$4		
 	
 	
 	
 !Q  ''11{$$W%,	 % 
 
 Hc3Z  ''
33H)))**11"*==
 	&$7		
 	
 	
 	
 	
 	
r6   c                    t          j        g dg dg                              t           j                  } t          j                            ddgddgdd	          }t          j        g d
g dg                              t           j                  t          j        g dg dg                              t           j                  g}t          || g|d           t          j        ddg                              t           j                  }t          j                            dddgddgd          }t          j        ddgddgg                              t           j                  t          j        g dg dg                              t           j                  g}t          || |g|d           d S )Nr   r8   r   r   r   r   r?   r!   rT   r@   rA   rB   rC   test_split_equal_parts_2drW   r"   r#   r   r   r   r9   r:   r$   rE   $test_split_variable_parts_2d_opset18rY   r0   s       r4   export_2d_opset18zSplit.export_2d_opset18   s   X+++-N-N-NO
 

&

 	 {$$9, % 
 
 Hooo788??
KKHooo'9'9'9:;;BB2:NN

 	<$,		
 	
 	
 	
 !Q  ''11{$$W%,	 % 
 
 HsCj3*-..55bjAAH***,C,C,CDEELL
 
 	&$7		
 	
 	
 	
 	
 	
r6   c                    t          j        g d                              t           j                  } t          j                            ddgg dd          }t          j        ddg                              t           j                  t          j        d	d
g                              t           j                  t          j        ddg                              t           j                  g}t          || g|d           t          j        ddg                              t           j                  }t          j                            dddgddg          }t          j        ddg                              t           j                  t          j        g d                              t           j                  g}t          || |g|d           d S )Nr   r   r   r   rS   r   r   rU   r   r   r   r   r   r   +test_split_equal_parts_default_axis_opset18rW   r!   r"   r#   r   r   rI   r$   .test_split_variable_parts_default_axis_opset18rY   r0   s       r4   export_default_values_opset18z#Split.export_default_values_opset18  s   X<<<==DDRZPP
 {$$9888	 % 
 
 Hc3Z  ''
33Hc3Z  ''
33Hc3Z  ''
33

 	<$>		
 	
 	
 	
 !Q  ''11{$$Wg.Z8P % 
 

 Hc3Z  ''
33H)))**11"*==
 	&$A		
 	
 	
 	
 	
 	
r6   c                 h   t          j        g                               t           j                  } t          j        g d                              t           j                  }t
          j                            dddgg d          }t          j        g                               t           j                  t          j        g                               t           j                  t          j        g                               t           j                  g}t          || |g|d           d S )	NrN   r   r   r#   r   rI   #test_split_zero_size_splits_opset18rW   )	r'   r(   r)   r*   r/   r+   r,   r-   r   rP   s       r4   export_zero_size_splits_opset18z%Split.export_zero_size_splits_opset18,  s    Xb\\((44
 ##**2844{$$W%888 % 
 
 HRLL
++HRLL
++HRLL
++

 	&$6		
 	
 	
 	
 	
 	
r6   c                 r   t          j        g d                              t           j                  } t          j                            ddgg dd          }t          j        ddg                              t           j                  t          j        d	d
g                              t           j                  t          j        ddg                              t           j                  t          j        dg                              t           j                  g}t          || g|d           d S )N)r   r   r   r   r   r   r9   r   r   )r   r   r   output_4r"   r`   r   r   r   r   r   r   r9   "test_split_1d_uneven_split_opset18rW   r'   r(   r)   r*   r+   r,   r-   r   r1   r2   r3   s      r4   export_1d_uneven_split_opset18z$Split.export_1d_uneven_split_opset18E  s   XAAABBII"*UU
 {$$9DDD	 % 
 
 Hc3Z  ''
33Hc3Z  ''
33Hc3Z  ''
33HcUOO""2:..	
 	<$5		
 	
 	
 	
 	
 	
r6   c                 2   t          j        g dg dg                              t           j                  } t          j                            ddgg ddd          }t          j        g d	g d
g                              t           j                  t          j        g dg dg                              t           j                  t          j        ddgddgg                              t           j                  g}t          || g|d           d S )N)r   r   r   r   r   r   r9   r:   )r;   r<   r=   r>         *@      ,@      .@      0@r   r   r   r?   rS   rT   r@   )r;   r<   r=   rB   )r>   rn   ro   r9   r:   rp   rq   "test_split_2d_uneven_split_opset18rW   rj   rk   s      r4   export_2d_uneven_split_opset18z$Split.export_2d_uneven_split_opset18^  s2   X888???
 

 &

 	 {$$9888 % 
 
 Hooo'8'8'89::AA"*MMHooo'9'9'9:;;BB2:NNHsCj4,/0077
CC
 	<$5		
 	
 	
 	
 	
 	
r6   N)r   r	   )__name__
__module____qualname__staticmethodr5   rG   rL   rQ   rZ   r^   rc   rf   rl   rs    r6   r4   r   r      sX       )
 )
 )
 \)
V +
 +
 +
 \+
Z $
 $
 $
 \$
L 
 
 
 \
2 (
 (
 (
 \(
T -
 -
 -
 \-
^ %
 %
 %
 \%
N 
 
 
 \
0 
 
 
 \
0 
 
 
 \
 
 
r6   r   )

__future__r   numpyr'   r+   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rx   r6   r4   <module>r}      s    # " " " " "      , , , , , , . . . . . .m
 m
 m
 m
 m
D m
 m
 m
 m
 m
r6   