convForward

wrapper of cudnnConvolutionForward for Variable

version(grain_cuda)

void

convForward

(

bool isConv

bool isNchw

T

size_t dim

size_t imDims

)

(

Variable!(T, dim, DeviceStorage) input

,

Variable!(T, dim, DeviceStorage) filter

,

ref Variable!(T, dim, DeviceStorage) output

,

const int[imDims] stride

,

const int[imDims] pad

,

const int[imDims] dilation

,

int ngroup = 1

,

cudnnConvolutionFwdAlgo_t algo = CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM

,

float alpha = 1

,

float beta = 0

)

Meta

Source

See Implementation