
    Si                        S SK Jr  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)    )annotations)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      Z/var/www/html/ai-image-ml/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*   r3   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   r:   r:   (   s    r   r:   c                      \ rS rSrSS jrSrg)Split_13,   Nc                $    U R                  XUS S9$ r1   r2   )r   r   r   r   s       r   r3   Split_13._run-   r5   r   r,   r6   r7   r,   r   r   r=   r=   ,   r8   r   r=   c                      \ rS rSrSS jrSrg)Split_181   Nc                "    U R                  XX4S9$ r1   r2   )r   r   r   r   r   s        r   r3   Split_18._run2   s    sNNr   r,   )NNNr7   r,   r   r   rB   rB   1   s    Or   rB   N)	
__future__r   onnx.reference.op_runr   r   r.   r:   r=   rB   r,   r   r   <module>rH      sO    # '% 4Hk H
	w 	H{ H
O{ Or   