Function rcudnn::cudnnGetConvolutionForwardWorkspaceSize
source · pub unsafe extern "C" fn cudnnGetConvolutionForwardWorkspaceSize(
handle: *mut cudnnContext,
xDesc: *mut cudnnTensorStruct,
wDesc: *mut cudnnFilterStruct,
convDesc: *mut cudnnConvolutionStruct,
yDesc: *mut cudnnTensorStruct,
algo: cudnnConvolutionFwdAlgo_t,
sizeInBytes: *mut usize
) -> cudnnStatus_t