
    h]                     ~    S SK Jr   " S S\5      r " S S\5      r " S S\5      r " S S	\5      r " S
 S\5      rg)    )OpRunc                        \ rS rSrS rS rSrg)CommonSplit	   c                 f    [         R                  " XU5        [        UR                  5      U l        g )N)r   __init__lenoutput	n_outputs)self	onnx_node
run_paramss      U/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_split.pyr   CommonSplit.__init__
   s"    t
3Y--.    c                    U=(       d    U R                   nUcp  UR                  U   U-  S:X  a  UR                  U   U-  nU/U-  nOAUR                  U   U-  S-   nU/U-  nUS==   UR                  U   [        U5      -
  -  ss'   UR                   Vs/ s H  n[        SU5      PM     nn/ n	Sn
U H4  n[        XU-   5      X'   X-  n
U	R	                  U[        U5         5        M6     [        U	5      $ s  snf )Nr      )r   shapesumsliceappendtuple)r   matsplitaxisnum_outputsr   divssliresposspls               r   
common_runCommonSplit.common_run   s    14>>	=yy*a/iio2	)iio2Q6	)b	SYYt_s5z99	$'II.IquQ{I.Cc9-CIJCJJs5:'  Sz /s   C8)r   N)__name__
__module____qualname____firstlineno__r   r$   __static_attributes__ r   r   r   r   	   s    /r   r   c                       \ rS rSrSS jrSrg)Split_2#   Nc                 $    U R                  XUS S9$ N)r   r   r$   )r   r   r   r   s       r   _runSplit_2._run$       s$GGr   r+   NNr&   r'   r(   r)   r2   r*   r+   r   r   r-   r-   #       Hr   r-   c                       \ rS rSrSrg)Split_11(   r+   N)r&   r'   r(   r)   r*   r+   r   r   r9   r9   (   s    r   r9   c                       \ rS rSrSS jrSrg)Split_13,   Nc                 $    U R                  XUS S9$ r0   r1   )r   r   r   r   s       r   r2   Split_13._run-   r4   r   r+   r5   r6   r+   r   r   r<   r<   ,   r7   r   r<   c                       \ rS rSrSS jrSrg)Split_181   Nc                 "    U R                  XX4S9$ r0   r1   )r   r   r   r   r   s        r   r2   Split_18._run2   s    sNNr   r+   )NNNr6   r+   r   r   rA   rA   1   s    Or   rA   N)onnx.reference.op_runr   r   r-   r9   r<   rA   r+   r   r   <module>rF      sL    (% 4Hk H
	w 	H{ H
O{ Or   