o
    toiE                     @  sD   d dl mZ d dlZd dlmZ G dd deZG dd deZdS )    )annotationsN)OpRunReduceNumpyc                   @  s   e Zd ZdddZdS )ReduceLogSum_1NTc                 C  sl   |d urt |nd }|jdkr| |tj ||S tj|||d}t|jdkr0tj||dfS t|fS Nr   )axiskeepdims)out)	tuplesizereduce_constantnpinfsumlenshapelog)selfdataaxesr   taxres r   X/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_reduce_log_sum.py_run   s   
zReduceLogSum_1._run)NT__name__
__module____qualname__r   r   r   r   r   r          r   c                   @  s   e Zd ZdddZdS )ReduceLogSum_18N   r   c                 C  sl   |  ||}|dk}|jdkr| |tj ||S tj|||d}t|jdkr0tj||dfS t|fS r   )	handle_axesr
   r   r   r   r   r   r   r   )r   r   r   r   noop_with_empty_axesr   r   r   r   r      s   
zReduceLogSum_18._run)Nr    r   r   r   r   r   r   r      r   r   )
__future__r   numpyr   onnx.reference.ops._opr   r   r   r   r   r   r   <module>   s
   