o
    wi                     @  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 )	GlobalAveragePool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_globalaveragepoolr   r   name)onnxhelper	make_nodenprandomrandnastypefloat32meantuplerangendimr   noder   r	    r$   j/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/globalaveragepool.pyexport   s    zGlobalAveragePool.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_globalaveragepool_precomputedr   )r   r   r   r   arrayr   r   r   r"   r$   r$   r%   $export_globalaveragepool_precomputed   s"   
z6GlobalAveragePool.export_globalaveragepool_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   