Function rcudnn::cudnnGetAttnDescriptor[][src]

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