Function rcudnn::cudnnSoftmaxBackward
source · pub unsafe extern "C" fn cudnnSoftmaxBackward(
handle: *mut cudnnContext,
algo: cudnnSoftmaxAlgorithm_t,
mode: cudnnSoftmaxMode_t,
alpha: *const c_void,
yDesc: *mut cudnnTensorStruct,
y: *const c_void,
dyDesc: *mut cudnnTensorStruct,
dy: *const c_void,
beta: *const c_void,
dxDesc: *mut cudnnTensorStruct,
dx: *mut c_void
) -> cudnnStatus_t