Function rcudnn::cudnnFindConvolutionForwardAlgorithmEx
source · pub unsafe extern "C" fn cudnnFindConvolutionForwardAlgorithmEx(
handle: *mut cudnnContext,
xDesc: *mut cudnnTensorStruct,
x: *const c_void,
wDesc: *mut cudnnFilterStruct,
w: *const c_void,
convDesc: *mut cudnnConvolutionStruct,
yDesc: *mut cudnnTensorStruct,
y: *mut c_void,
requestedAlgoCount: i32,
returnedAlgoCount: *mut i32,
perfResults: *mut cudnnConvolutionFwdAlgoPerfStruct,
workSpace: *mut c_void,
workSpaceSizeInBytes: usize
) -> cudnnStatus_t