U ‰dœã@s"d dd„Zd dd„Zd dd„ZdS) FÚNCHWcKs&|r d|d<|jj||fd|i|—ŽS)z Max poolingÚCUDNNÚengineÚorder)ÚnetZMaxPool©ÚmodelÚblob_inÚblob_outZ use_cudnnrÚkwargs©r úA/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/helpers/pooling.pyÚmax_pool sr cKs&|r d|d<|jj||fd|i|—ŽS)zAverage poolingrrr)rZ AveragePoolrr r r Ú average_poolsþýürcKs&|jj|||dgfd|i|—ŽdS)z2Max pooling with an explicit index of max positionÚ_indexré)rZMaxPoolWithIndex)rrr rr r r r Úmax_pool_with_indexs þýüûrN)Fr)Fr)r)r rrr r r r Ú s