
    /j                    ^    d dl mZ d dlZd dlmZ  G d de          Z G d de          ZdS )    )annotationsN)OpRunReduceNumpyc                      e Zd ZddZdS )ReduceSum_1Nc                    |t          |          nd }t          j        ||||j                  }|dk    r.t	          |t          j                  st          j        |          }|fS )Naxiskeepdimsdtyper   )tuplenpsumr   
isinstancendarrayarray)selfxaxesr
   ress        e/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/reference/ops/op_reduce_sum.py_runzReduceSum_1._run   s]    ".uT{{{DfQTHAGDDDq==C!<!<=(3--Cv    )NN__name__
__module____qualname__r    r   r   r   r      s(             r   r   c                      e Zd ZddZdS )ReduceSum_13Nc           
     :   |                      ||          }	 t          j        ||||j                  }|dk    r.t	          |t          j                  st          j        |          }|fS # t          $ r$}t          d|j        d|d| d          |d }~ww xY w)Nr   r   zUnable to reduce shape z with axes=z and keepdims=.)	handle_axesr   r   r   r   r   r   	TypeErrorshape)r   r   r   r
   noop_with_empty_axesr   es          r   r   zReduceSum_13._run   s    &:;;
	&HHHC1}}ZRZ%@%@}hsmm 6M  	 	 	a!'aaaaV^aaa 	s   AA, ,
B6BB)NNNr   r   r   r   r   r      s(             r   r   )
__future__r   numpyr   onnx.reference.ops._opr   r   r   r   r   r   <module>r*      s    # " " " " "     3 3 3 3 3 3    "       #     r   