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 )ReduceMax_1Nc                 C  s   |d k	rt |nd }|jdkrTt|jtjr<t|jjntj }| 	||||S tj
j|||dkd}|dkrt|tjst|}|fS )Nr      Zaxiskeepdims)tuplesizenp
issubdtypedtypeintegeriinfomininfreduce_constantmaximumreduce
isinstancendarrayarray)selfdataaxesr   minvalueres r   D/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_reduce_max.py_run   s    

zReduceMax_1._run)NN__name__
__module____qualname__r   r   r   r   r   r      s   r   c                   @  s   e Zd ZddddddZdS )	ReduceMax_18Nr   r   int)r   noop_with_empty_axesc                 C  s   |  |r|dkr|fS | |}|dk}|jdkrjt|jtjrRt|jjntj	 }| 
||||S tjj|||d}|dkrt|tjst|}|fS )Nr   r   )Zis_axes_emptyZhandle_axesr	   r
   r   r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   r%   r   r   r   r   r   r      s    


zReduceMax_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   