U
    qh                     @  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 )ReduceSumSquare_1Nc                 C  sN   |d k	rt |nd }tjt|||d}|dkrHt|tjsHt|}|fS )NZaxiskeepdimsr   )tuplenpsumsquare
isinstancendarrayarray)selfdataaxesr   res r   K/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_reduce_sum_square.py_run   s
    
zReduceSumSquare_1._run)NN__name__
__module____qualname__r   r   r   r   r   r      s   r   c                   @  s   e Zd ZdddZdS )ReduceSumSquare_18N   r   c                 C  sj   |  |r|dkrt|fS | |}|dk}tjt|||d}|dkrdt|tjsdt|}|fS )Nr   r   )Zis_axes_emptyr   r
   Zhandle_axesr	   r   r   r   )r   r   r   r   Znoop_with_empty_axesr   r   r   r   r      s    

zReduceSumSquare_18._run)Nr   r   r   r   r   r   r   r      s   r   )
__future__r   Znumpyr   Zonnx.reference.ops._opr   r   r   r   r   r   r   <module>   s   
