U
    rh                     @  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                   @  s0   e Zd ZeddddZeddddZdS )GlobalMaxPoolNone)returnc                  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)ZaxisZkeepdimsZtest_globalmaxpoolr   r   name)onnxhelper	make_nodenprandomZrandnastypefloat32maxtuplerangendimr   noder   r	    r    M/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/globalmaxpool.pyexport   s     zGlobalMaxPool.exportc                  C  sv   t jjddgdgd} tdddgdd	d
gdddg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            	   Ztest_globalmaxpool_precomputedr   )r   r   r   r   arrayr   r   r   r   r    r    r!    export_globalmaxpool_precomputed   s"    
z.GlobalMaxPool.export_globalmaxpool_precomputedN)__name__
__module____qualname__staticmethodr"   r)   r    r    r    r!   r      s   
r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r    r    r    r!   <module>   s
   