Class PoolingWrapper

• Defined in File cudnn_wrappers.h

Inheritance Relationships

Base Type

• public marian::CUDNNWrapper (Class CUDNNWrapper)

Class Documentation

class PoolingWrapper : public marian::CUDNNWrapper

Public Functions

PoolingWrapper(int height, int width, int padHeight, int padWidth, int strideHeight, int strideWidth, std::string mode)

void getOutputShape(const Shape &xShape, Shape &shape)

void forward(Tensor x, Tensor y)

void backward(Tensor x, Tensor xGrad, Tensor y, Tensor yGrad)

~PoolingWrapper()