o
    wi                     @  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 urt |nd }|jdkr*t|jtjrt|jjntj }| 	||||S tj
j|||dkd}|dkrDt|tjsDt|}|fS )Nr      axiskeepdims)tuplesizenp
issubdtypedtypeintegeriinfomininfreduce_constantmaximumreduce
isinstancendarrayarray)selfdataaxesr   minvalueres r   ]/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/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ZdS )ReduceMax_18Nr   r   r   intnoop_with_empty_axesc                 C  s   |  |r|dkr|fS | |}|dk}|jdkr5t|jtjr)t|jjntj	 }| 
||||S tjj|||d}|dkrMt|tjsMt|}|fS )Nr   r   )is_axes_empty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   r   r   r   r$      s    r$   )
__future__r   numpyr   onnx.reference.ops._opr   r   r$   r   r   r   r   <module>   s
   