poolBackward

wrapper of cudnnPoolingBackward for Variable

version(grain_cuda)

void

poolBackward

(

bool isMax = true

bool isAveragePad = false

T

size_t _tensorDims

size_t _poolDims

)

(

ref Variable!(T, _tensorDims, DeviceStorage) gradInput

,

Variable!(T, _tensorDims, DeviceStorage) input

,

Variable!(T, _tensorDims, DeviceStorage) gradOutput

,

Variable!(T, _tensorDims, DeviceStorage) output

,

int[_poolDims] windowDim

,

int[_poolDims] padding

,

int[_poolDims] stride

,

T alpha = 1

,

T beta = 0

)

Meta

Source

See Implementation