Function rcudnn::cudnnMultiHeadAttnBackwardWeights[][src]

pub unsafe extern "C" fn cudnnMultiHeadAttnBackwardWeights(
    handle: *mut cudnnContext,
    attnDesc: *mut cudnnAttnStruct,
    addGrad: cudnnWgradMode_t,
    qDesc: *mut cudnnSeqDataStruct,
    queries: *const c_void,
    kDesc: *mut cudnnSeqDataStruct,
    keys: *const c_void,
    vDesc: *mut cudnnSeqDataStruct,
    values: *const c_void,
    doDesc: *mut cudnnSeqDataStruct,
    dout: *const c_void,
    weightSizeInBytes: usize,
    weights: *const c_void,
    dweights: *mut c_void,
    workSpaceSizeInBytes: usize,
    workSpace: *mut c_void,
    reserveSpaceSizeInBytes: usize,
    reserveSpace: *mut c_void
) -> cudnnStatus_t