Function average_pool

caffe2/python/helpers/pooling.py:18–28

Average pooling

(model, blob_in, blob_out, use_cudnn=False, order="NCHW",
                 **kwargs)

def average_pool(model, blob_in, blob_out, use_cudnn=False, order="NCHW",
                 **kwargs):
    """Average pooling"""
    if use_cudnn:
        kwargs['engine'] = 'CUDNN'
    return model.net.AveragePool(
        blob_in,
        blob_out,
        order=order,
        **kwargs
    )


def max_pool_with_index(model, blob_in, blob_out, order="NCHW", **kwargs):