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