
    wi                        d dl mZ d dlmZ  G d de          Z G d de          Z G d de          Z G d	 d
e          Z G d de          ZdS )    )annotations)OpRunc                      e Zd Zd Zd ZdS )CommonSplitc                d    t          j        | ||           t          |j                  | _        d S )N)r   __init__lenoutput	n_outputs)self	onnx_node
run_paramss      k/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/onnx/reference/ops/op_split.pyr   zCommonSplit.__init__
   s,    tY
333Y-..    c                   |p| j         }|o|j        |         |z  dk    r|j        |         |z  }|g|z  }nD|j        |         |z  dz   }|g|z  }|dxx         |j        |         t          |          z
  z  cc<   d |j        D             }g }d}	|D ]E}
t          |	|	|
z             ||<   |	|
z  }	|                    |t          |                              Ft          |          S )Nr      c                .    g | ]}t          d |          S )r   )slice).0ss     r   
<listcomp>z*CommonSplit.common_run.<locals>.<listcomp>   s     ...quQ{{...r   )r   shapesumr   appendtuple)r   matsplitaxisnum_outputsr   divsliresposspls              r   
common_runzCommonSplit.common_run   s   14>	=y*a//io2	)io2Q6	)b			SYt_s5zz99			..CI... 	( 	(Cc39--CI3JCJJs5::''''Szzr   N)__name__
__module____qualname__r   r&    r   r   r   r   	   s2        / / /    r   r   c                      e Zd ZddZdS )Split_2Nc                4    |                      |||d           S N)r   r    r&   )r   r   r   r   s       r   _runzSplit_2._run$       sE$GGGr   NNr'   r(   r)   r0   r*   r   r   r,   r,   #   .        H H H H H Hr   r,   c                      e Zd ZdS )Split_11N)r'   r(   r)   r*   r   r   r6   r6   (   s        Dr   r6   c                      e Zd ZddZdS )Split_13Nc                4    |                      |||d           S r.   r/   )r   r   r   r   s       r   r0   zSplit_13._run-   r1   r   r2   r3   r*   r   r   r8   r8   ,   r4   r   r8   c                      e Zd ZddZdS )Split_18Nc                4    |                      ||||          S r.   r/   )r   r   r   r   r    s        r   r0   zSplit_18._run2   s    sE+NNNr   )NNNr3   r*   r   r   r;   r;   1   s.        O O O O O Or   r;   N)	
__future__r   onnx.reference.op_runr   r   r,   r6   r8   r;   r*   r   r   <module>r?      s!   # " " " " " ' ' ' ' ' '    %   4H H H H Hk H H H
	 	 	 	 	w 	 	 	H H H H H{ H H H
O O O O O{ O O O O Or   