
    bIgE.                    N    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	y)    )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y)Splitc            
     6   t        j                  g d      j                  t         j                        } t        j
                  j                  ddgg dd      }t        j                  ddg      j                  t         j                        t        j                  d	d
g      j                  t         j                        t        j                  ddg      j                  t         j                        g}t        || g|dt        j
                  j                  dd      g       t        j                  ddg      j                  t         j                        }t        j
                  j                  dddgddgd      }t        j                  ddg      j                  t         j                        t        j                  g d      j                  t         j                        g}t        || |g|dt        j
                  j                  dd      g       y )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/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/split.pyexport_1d_opset13zSplit.export_1d_opset13   s   XX<=DDRZZP
{{$$98	 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$4;;33B;<	
 !Q ''1{{$$W%,	 % 
 HHc3Z ''

3HH)*11"**=
 	&$7;;33B;<	
    c            
        t        j                  g dg dg      j                  t         j                        } t        j
                  j                  ddgddgd      }t        j                  g d	g d
g      j                  t         j                        t        j                  g dg dg      j                  t         j                        g}t        || g|dt        j
                  j                  dd      g       t        j                  ddg      j                  t         j                        }t        j
                  j                  dddgddgd      }t        j                  ddgddgg      j                  t         j                        t        j                  g dg dg      j                  t         j                        g}t        || |g|dt        j
                  j                  dd      g       y )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_opset13:   s   XX+-NO

&
 	 {{$$WI
J/Ga % 

 HHo78??

KHHo'9:;BB2::N

 	<$4;;33B;<	
 !Q ''1{{$$W%,	 % 
 HHsCj3*-.55bjjAHH*,CDELL


 	&$7;;33B;<	
r4   c            
     2   t        j                  g d      j                  t         j                        } t        j
                  j                  ddgg d      }t        j                  ddg      j                  t         j                        t        j                  dd	g      j                  t         j                        t        j                  d
dg      j                  t         j                        g}t        || g|dt        j
                  j                  dd      g       t        j                  ddg      j                  t         j                        }t        j
                  j                  dddgddg      }t        j                  ddg      j                  t         j                        t        j                  g d      j                  t         j                        g}t        || |g|dt        j
                  j                  dd      g       y )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_opset13h   s   XX<=DDRZZP
 {{$$WI/S % 

 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$>;;33B;<	
 !Q ''1{{$$Wg.Z8P % 

 HHc3Z ''

3HH)*11"**=
 	&$A;;33B;<	
r4   c            
        t        j                  g       j                  t         j                        } t        j                  g d      j                  t         j                        }t
        j                  j                  dddgg d      }t        j                  g       j                  t         j                        t        j                  g       j                  t         j                        t        j                  g       j                  t         j                        g}t        || |g|dt
        j                  j                  dd	      g
       y )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     XXb\((4
 #**2884{{$$W%8 % 
 HHRL

+HHRL

+HHRL

+

 	&$6;;33B;<	
r4   c                    t        j                  g d      j                  t         j                        } t        j
                  j                  ddgg ddd      }t        j                  dd	g      j                  t         j                        t        j                  d
dg      j                  t         j                        t        j                  ddg      j                  t         j                        g}t        || g|d       t        j                  ddg      j                  t         j                        }t        j
                  j                  dddgddgd      }t        j                  dd	g      j                  t         j                        t        j                  g d      j                  t         j                        g}t        || |g|d       y )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{   XX<=DDRZZP
{{$$98 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$4		
 !Q ''1{{$$W%,	 % 
 HHc3Z ''

3HH)*11"**=
 	&$7		
r4   c                 z   t        j                  g dg dg      j                  t         j                        } t        j
                  j                  ddgddgdd	      }t        j                  g d
g dg      j                  t         j                        t        j                  g dg dg      j                  t         j                        g}t        || g|d       t        j                  ddg      j                  t         j                        }t        j
                  j                  dddgddgd      }t        j                  ddgddgg      j                  t         j                        t        j                  g dg dg      j                  t         j                        g}t        || |g|d       y )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   XX+-NO

&
 	 {{$$9, % 
 HHo78??

KHHo'9:;BB2::N

 	<$,		
 !Q ''1{{$$W%,	 % 
 HHsCj3*-.55bjjAHH*,CDELL


 	&$7		
r4   c                    t        j                  g d      j                  t         j                        } t        j
                  j                  ddgg dd      }t        j                  ddg      j                  t         j                        t        j                  d	d
g      j                  t         j                        t        j                  ddg      j                  t         j                        g}t        || g|d       t        j                  ddg      j                  t         j                        }t        j
                  j                  dddgddg      }t        j                  ddg      j                  t         j                        t        j                  g d      j                  t         j                        g}t        || |g|d       y )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  su   XX<=DDRZZP
 {{$$98	 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3

 	<$>		
 !Q ''1{{$$Wg.Z8P % 

 HHc3Z ''

3HH)*11"**=
 	&$A		
r4   c                 h   t        j                  g       j                  t         j                        } t        j                  g d      j                  t         j                        }t
        j                  j                  dddgg d      }t        j                  g       j                  t         j                        t        j                  g       j                  t         j                        t        j                  g       j                  t         j                        g}t        || |g|d       y )	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     XXb\((4
 #**2884{{$$W%8 % 
 HHRL

+HHRL

+HHRL

+

 	&$6		
r4   c                 r   t        j                  g d      j                  t         j                        } t        j
                  j                  ddgg dd      }t        j                  ddg      j                  t         j                        t        j                  d	d
g      j                  t         j                        t        j                  ddg      j                  t         j                        t        j                  dg      j                  t         j                        g}t        || g|d       y )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_opset18E  s    XXABII"**U
 {{$$9D	 % 
 HHc3Z ''

3HHc3Z ''

3HHc3Z ''

3HHcUO""2::.	
 	<$5		
r4   c                 0   t        j                  g dg dg      j                  t         j                        } t        j
                  j                  ddgg ddd      }t        j                  g d	g d
g      j                  t         j                        t        j                  g dg dg      j                  t         j                        t        j                  ddgddgg      j                  t         j                        g}t        || g|d       y )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^  s    XX8?

 &
 	 {{$$98 % 
 HHo'89:AA"**MHHo'9:;BB2::NHHsCj4,/077

C
 	<$5		
r4   N)returnNone)__name__
__module____qualname__staticmethodr3   rE   rJ   rO   rX   r\   ra   rd   rj   rq    r4   r2   r   r      s    )
 )
V +
 +
Z $
 $
L 
 
2 (
 (
T -
 -
^ %
 %
N 
 
0 
 
0 
 
r4   r   )

__future__r   numpyr%   r)   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rx   r4   r2   <module>r}      s$    #   , .m
D m
r4   