Skip to main content

Crate dlpack_ffi

Crate dlpack_ffi 

Source

Structs§

DLDataType
\brief The data type the tensor can hold. The data type is assumed to follow the native endian-ness. An explicit error message should be raised when attempting to export an array with non-native endianness
DLDevice
\brief A Device for Tensor and operator.
DLManagedTensor
\brief C Tensor object, manage memory of DLTensor. This data structure is intended to facilitate the borrowing of DLTensor by another framework. It is not meant to transfer the tensor. When the borrowing framework doesn’t need the tensor, it should call the deleter to notify the host that the resource is no longer needed.
DLManagedTensorVersioned
\brief A versioned and managed C Tensor object, manage memory of DLTensor.
DLPackExchangeAPI
\brief Framework-specific function pointers table for DLPack exchange.
DLPackExchangeAPIHeader
\brief DLPackExchangeAPI stable header. \sa DLPackExchangeAPI
DLPackVersion
\brief The DLPack version.
DLTensor
\brief Plain C Tensor object, does not manage memory.

Enums§

DLDataTypeCode
\brief The type code options DLDataType.
DLDeviceType

Constants§

DLPACK_FLAG_BITMASK_IS_COPIED
DLPACK_FLAG_BITMASK_IS_SUBBYTE_TYPE_PADDED
DLPACK_FLAG_BITMASK_READ_ONLY
DLPACK_MAJOR_VERSION
DLPACK_MINOR_VERSION

Type Aliases§

DLPackCurrentWorkStream
\brief Obtain the current work stream of a device.
DLPackDLTensorFromPyObjectNoSync
\brief Exports a PyObject* Tensor/NDArray to a provided DLTensor.
DLPackManagedTensorAllocator
\brief Request a producer library to create a new tensor.
DLPackManagedTensorFromPyObjectNoSync
\brief Exports a PyObject* Tensor/NDArray to a DLManagedTensorVersioned.
DLPackManagedTensorToPyObjectNoSync
\brief Imports a DLManagedTensorVersioned to a PyObject* Tensor/NDArray.