convBackward

wrapper of cudnnConvolutionBackwardData and Weight for Variable

version(grain_cuda)

void

convBackward

(

bool isConv

bool isNchw

T

size_t dim

size_t imDims

)

(

ref Variable!(T, dim, DeviceStorage) gradInput

,

Variable!(T, dim, DeviceStorage) input

,

ref Variable!(T, dim, DeviceStorage) gradFilter

,

Variable!(T, dim, DeviceStorage) filter

,

Variable!(T, dim, DeviceStorage) gradOutput

,

const int[imDims] stride

,

const int[imDims] pad

,

const int[imDims] dilation

,

int ngroup = 1

,

cudnnConvolutionBwdDataAlgo_t algo = CUDNN_CONVOLUTION_BWD_DATA_ALGO_1

,

float alpha = 1

,

float beta = 0

)

Meta

Source

See Implementation