
    /j                    V    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e          Z	dS )    )annotationsN)Base)expectc                  >    e Zd Zedd            Zedd            ZdS )GlobalAveragePoolreturnNonec                 ~   t           j                            ddgdg          } t          j                            dddd                              t          j                  }t          j        |t          t          dt          j        |                              d	
          }t          | |g|g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   s      r/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/globalaveragepool.pyexportzGlobalAveragePool.export   s    {$$5E % 
 

 IOOAq!Q''..rz::GAE%271::"6"677$GGGtQC!3KLLLLLL    c                 Z   t           j                            ddgdg          } t          j        g dg dg dggg                              t          j                  }t          j        dgggg                              t          j                  }t          | |g|gd	
           d S )Nr   r   r   r   )r   r   r   )   r      )      	   r   "test_globalaveragepool_precomputedr   )r   r   r   r   arrayr   r    r   r%   s      r'   $export_globalaveragepool_precomputedz6GlobalAveragePool.export_globalaveragepool_precomputed   s    {$$5E % 
 

 H "		!		!		

 

 &

 	
 HugY&&rz22tQC!3WXXXXXXr)   N)r   r	   )__name__
__module____qualname__staticmethodr(   r2    r)   r'   r   r      s\        M M M \M Y Y Y \Y Y Yr)   r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r7   r)   r'   <module>r<      s    # " " " " "      , , , , , , . . . . . .Y Y Y Y Y Y Y Y Y Yr)   