Function rcudnn_sys::cudnnGetAttnDescriptor[][src]

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