
    Ng".                     J    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 )    N)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returnNc            
      :   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       ]/var/www/html/ai-engine/env/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   r7   r8   r9   r   r   r   r:   r;   r<   !test_split_equal_parts_2d_opset13r   r   r   r   r    r!   r
   r   r7   r8   r"   r9   r:   r;   r<   $test_split_variable_parts_2d_opset13r$   r.   s       r2   export_2d_opset13zSplit.export_2d_opset139   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;;<	
 	
 	
 	
 	
 	
r4   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$   r.   s       r2   export_default_values_opset13z#Split.export_default_values_opset13g   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;;<	
 	
 	
 	
 	
 	
r4   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   rG   #test_split_zero_size_splits_opset13r   r   r   )
r%   r&   r'   r(   r-   r)   r*   r+   r   r,   r/   r!   r0   r1   s       r2   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;;<	
 	
 	
 	
 	
 	
r4   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-   r.   s       r2   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		
 	
 	
 	
 	
 	
r4   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	   r6   r   r   r   r   r=   r   rR   r>   r?   r@   rA   test_split_equal_parts_2drU   r    r!   r   r
   r   r7   r8   r"   rC   $test_split_variable_parts_2d_opset18rW   r.   s       r2   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		
 	
 	
 	
 	
 	
r4   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   rQ   r   r   rS   r
   r   r   r   r   r   +test_split_equal_parts_default_axis_opset18rU   r   r    r!   r   r   rG   r"   .test_split_variable_parts_default_axis_opset18rW   r.   s       r2   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		
 	
 	
 	
 	
 	
r4   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 )	NrL   r   r   r!   r   rG   #test_split_zero_size_splits_opset18rU   )	r%   r&   r'   r(   r-   r)   r*   r+   r   rN   s       r2   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		
 	
 	
 	
 	
 	
r4   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   r7   r   r   )r   r   r   output_4r    r^   r
   r   r   r   r   r   r7   "test_split_1d_uneven_split_opset18rU   r%   r&   r'   r(   r)   r*   r+   r   r/   r0   r1   s      r2   export_1d_uneven_split_opset18z$Split.export_1d_uneven_split_opset18D  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		
 	
 	
 	
 	
 	
r4   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   r7   r8   )r9   r:   r;   r<         *@      ,@      .@      0@r   r   r   r=   rQ   rR   r>   )r9   r:   r;   r@   )r<   rl   rm   r7   r8   rn   ro   "test_split_2d_uneven_split_opset18rU   rh   ri   s      r2   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		
 	
 	
 	
 	
 	
r4   )r   N)__name__
__module____qualname__staticmethodr3   rE   rJ   rO   rX   r\   ra   rd   rj   rq    r4   r2   r   r      sX       )
 )
 )
 \)
V +
 +
 +
 \+
Z $
 $
 $
 \$
L 
 
 
 \
2 (
 (
 (
 \(
T -
 -
 -
 \-
^ %
 %
 %
 \%
N 
 
 
 \
0 
 
 
 \
0 
 
 
 \
 
 
r4   r   )numpyr%   r)   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rv   r4   r2   <module>rz      s~   
      , , , , , , . . . . . .m
 m
 m
 m
 m
D m
 m
 m
 m
 m
r4   