o
    toi                     @  s>   d dl mZ d dlZd dlmZ dddZG d	d
 d
eZdS )    )annotationsN)OpRunx
np.ndarrayreturnc                 C  sF   t | d }| jtt||d d}t|D ]}t |d}q|S )N   )axis)npndimmaxtuplerangeexpand_dims)r   spatial_shapey_ r   Y/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_global_max_pool.py_global_max_pool   s
   r   c                   @  s   e Zd Zdd ZdS )GlobalMaxPoolc                 C  s   t |}|fS )N)r   )selfr   resr   r   r   _run   s   zGlobalMaxPool._runN)__name__
__module____qualname__r   r   r   r   r   r      s    r   )r   r   r   r   )
__future__r   numpyr
   onnx.reference.op_runr   r   r   r   r   r   r   <module>   s
   
