U
    qh                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)
CommonPoolc                   @  s   e Zd ZdddZdS )LpPoolN   c
                 C  sX   t j| d|	tt||||||||d
}
t|}t||
d  d| |jfS )NZAVG)auto_pad	ceil_mode	dilationskernel_shapepadsstridesr   g      ?)r   _runnppowerabsoluteprodZastypeZdtype)selfxr   r   r   r	   pr
   r   Zcount_include_padZpower_averageZkernel_element_count r   A/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_lp_pool.pyr      s    
zLpPool._run)NNNNr   NNN)__name__
__module____qualname__r   r   r   r   r   r      s           r   )
__future__r   Znumpyr   Z!onnx.reference.ops.op_pool_commonr   r   r   r   r   r   <module>   s   