Function rcudnn_sys::cudnnGetMultiHeadAttnBuffers [−][src]
pub unsafe extern "C" fn cudnnGetMultiHeadAttnBuffers(
handle: cudnnHandle_t,
attnDesc: cudnnAttnDescriptor_t,
weightSizeInBytes: *mut usize,
workSpaceSizeInBytes: *mut usize,
reserveSpaceSizeInBytes: *mut usize
) -> cudnnStatus_t