o
    soi                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s(   e Zd ZedddZedddZdS )	GlobalMaxPoolreturnNonec                  C  sh   t jjddgdgd} tjddddtj}tj|t	t
dt|d	d
}t| |g|gdd d S )Nr   xyinputsoutputs            T)axiskeepdimstest_globalmaxpoolr   r   name)onnxhelper	make_nodenprandomrandnastypefloat32maxtuplerangendimr   noder   r	    r$   ]/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/globalmaxpool.pyexport   s    zGlobalMaxPool.exportc                  C  sp   t jjddgdgd} tg dg dg dgggtj}tdggggtj}t| |g|gd	d
 d S )Nr   r   r	   r
   )r   r   r   )   r      )      	   r+   test_globalmaxpool_precomputedr   )r   r   r   r   arrayr   r   r   r"   r$   r$   r%    export_globalmaxpool_precomputed   s"   
z.GlobalMaxPool.export_globalmaxpool_precomputedN)r   r   )__name__
__module____qualname__staticmethodr&   r.   r$   r$   r$   r%   r      s
    
r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r$   r$   r$   r%   <module>   s   