
    
Ngt                     6    d dl Zd dlmZ  G d de          ZdS )    N)OpRunc                       e Zd ZddddZdS )CumSumN)	exclusivereversec                   |sd n|d         }|(|s|rt          d          t          j        |          fS t          |t          j        t          j        f          s|t          |j                  dk    s)t          |j                  dk    r,|j        d         dk    rt          d| d|j         d          t          |j                  dk    r|d         }|r:d |j        D             }t          d d d          ||<   |t          |                   }|rd	 |j        D             }d
 |j        D             }t          dd          ||<   t          d|j        |                   ||<   t          j        |j        |j                  }t          j        |t          |                   ||t          |                              nt          j        ||          }|r|t          |                   }|fS )Nr   z(reverse=1 or exclusive=1 not implemented   z(axis must be an array of one number not z (shape z).c                 .    g | ]}t          d |          S r   slice.0ss     Y/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_cum_sum.py
<listcomp>zCumSum._run.<locals>.<listcomp>   s     88815A;;888    c                 .    g | ]}t          d |          S r   r   r   s     r   r   zCumSum._run.<locals>.<listcomp>        666q!666r   c                 .    g | ]}t          d |          S r   r   r   s     r   r   zCumSum._run.<locals>.<listcomp>   r   r   )dtype)axisout)r   )NotImplementedErrornpcumsum
isinstanceint32int64lenshapeRuntimeErrorr   tuplezerosr   )	selfxr   r   r   rev_indices	indices_c	indices_dress	            r   _runzCumSum._run   s   ,ttT!W< V) V)*TUUUIaLL?"$28 455 	4:""s4:':':tz!}PQ?Q?Q"[t[[TZ[[[   4:""Aw 	&88888K %dD" 5 5K%$$%A 	*66ag666I66ag666I#ArllIdO#Aqwt}55IdO(17!'222CIai(()#eI>N>N:OPPPPP)AD)))C 	*eK(()Cvr   )__name__
__module____qualname__r,    r   r   r   r      s2        '+T       r   r   )numpyr   onnx.reference.op_runr   r   r0   r   r   <module>r3      sY        ' ' ' ' ' '    U     r   