Function rcudnn::cudnnSetAttnDescriptor[][src]

pub unsafe extern "C" fn cudnnSetAttnDescriptor(
    attnDesc: *mut cudnnAttnStruct,
    attnMode: u32,
    nHeads: i32,
    smScaler: f64,
    dataType: cudnnDataType_t,
    computePrec: cudnnDataType_t,
    mathType: cudnnMathType_t,
    attnDropoutDesc: *mut cudnnDropoutStruct,
    postDropoutDesc: *mut cudnnDropoutStruct,
    qSize: i32,
    kSize: i32,
    vSize: i32,
    qProjSize: i32,
    kProjSize: i32,
    vProjSize: i32,
    oProjSize: i32,
    qoMaxSeqLength: i32,
    kvMaxSeqLength: i32,
    maxBatchSize: i32,
    maxBeamSize: i32
) -> cudnnStatus_t