Function rcudnn::cudnnGetMultiHeadAttnWeights[][src]

pub unsafe extern "C" fn cudnnGetMultiHeadAttnWeights(
    handle: *mut cudnnContext,
    attnDesc: *mut cudnnAttnStruct,
    wKind: cudnnMultiHeadAttnWeightKind_t,
    weightSizeInBytes: usize,
    weights: *const c_void,
    wDesc: *mut cudnnTensorStruct,
    wAddr: *mut *mut c_void
) -> cudnnStatus_t