Crate dlpark

Modules

• ffi
  Raw bindings for DLPack.
• prelude
  Imports the structs and traits for you to implement IntoDLPack and FromDLPack.
• utils

Structs

• ManagedTensor
  Safe wrapper for DLManagedTensor. Will call deleter when dropped.
• ManagerCtx
  The ManagerCtx holds the Tensor and its metadata.

Enums

• ShapeAndStrides
  If the shape or strides of Tensor is vec of i64, then it should be borrowed to avoid copy. The lifetime should be ’static since we don’t managed its memory. Otherwise, we should copy the data and convert its type to i64 and managed it ourselves.

Traits

• FromDLPack
  Make Tensor from DLPack
• InferDtype
  Infer DataType from generic parameter.
• IntoDLPack
  Convert into DLPack
• TensorView
  Access Tensor data.
• ToTensor
  User should implement this trait for their tensor.

Type Aliases

• DLPack
  DLPack is a data structure that can be used to describe tensor data. It’s a pointer to a DLManagedTensor.