
    
Ng]                         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 )    )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      W/var/www/html/ai-engine/env/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(   r/   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   r5   r5   (   s        Dr   r5   c                       e Zd ZddZdS )Split_13Nc                 4    |                      |||d           S r-   r.   )r   r   r   r   s       r   r/   zSplit_13._run-   r0   r   r1   r2   r)   r   r   r7   r7   ,   r3   r   r7   c                       e Zd ZddZdS )Split_18Nc                 4    |                      ||||          S r-   r.   )r   r   r   r   r   s        r   r/   zSplit_18._run2   s    sE+NNNr   )NNNr2   r)   r   r   r:   r:   1   s.        O O O O O Or   r:   N)onnx.reference.op_runr   r   r+   r5   r7   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   