a
    ==icœ  ã                   @   s"   d	dd„Z d
dd„Zddd„ZdS )FÚNCHWc                 K   s&   |rd|d< | j j||fd|i|¤ŽS )zMax poolingÚCUDNNÚengineÚorder)ÚnetZMaxPool©ÚmodelÚblob_inÚblob_outZ	use_cudnnr   Úkwargs© r   ún/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/helpers/pooling.pyÚmax_pool   s    r   c                 K   s&   |rd|d< | j j||fd|i|¤ŽS )zAverage poolingr   r   r   )r   ZAveragePoolr   r   r   r   Úaverage_pool   s    þýür   c                 K   s&   | j j|||d gfd|i|¤Žd S )z2Max pooling with an explicit index of max positionÚ_indexr   é    )r   ZMaxPoolWithIndex)r   r   r	   r   r
   r   r   r   Úmax_pool_with_index   s    
þýüûr   N)Fr   )Fr   )r   )r   r   r   r   r   r   r   Ú<module>   s   

