Function rcudnn::cudnnMultiHeadAttnBackwardData[][src]

pub unsafe extern "C" fn cudnnMultiHeadAttnBackwardData(
    handle: *mut cudnnContext,
    attnDesc: *mut cudnnAttnStruct,
    loWinIdx: *const i32,
    hiWinIdx: *const i32,
    devSeqLengthsDQDO: *const i32,
    devSeqLengthsDKDV: *const i32,
    doDesc: *mut cudnnSeqDataStruct,
    dout: *const c_void,
    dqDesc: *mut cudnnSeqDataStruct,
    dqueries: *mut c_void,
    queries: *const c_void,
    dkDesc: *mut cudnnSeqDataStruct,
    dkeys: *mut c_void,
    keys: *const c_void,
    dvDesc: *mut cudnnSeqDataStruct,
    dvalues: *mut c_void,
    values: *const c_void,
    weightSizeInBytes: usize,
    weights: *const c_void,
    workSpaceSizeInBytes: usize,
    workSpace: *mut c_void,
    reserveSpaceSizeInBytes: usize,
    reserveSpace: *mut c_void
) -> cudnnStatus_t