convForward

wrapper of cudnnConvolutionForward for Variable

version(grain_cuda)
void
convForward
(
bool isConv
bool isNchw
T
size_t dim
size_t imDims
)
(,,,
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