dlpark 0.8.0-alpha.3

dlpack Rust binding for Python
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

//! Conversions between DLPack and supported container libraries.
//!
//! Producer-side conversions return [`crate::Builder`] so callers can adjust
//! flags and metadata before selecting a managed tensor ABI.

#[cfg(feature = "cudarc")]
/// CUDA device-buffer interop through `cudarc`.
pub mod cudarc;

#[cfg(feature = "image")]
/// HWC image interop through the `image` crate.
pub mod image;

#[cfg(feature = "ndarray")]
/// Dynamic-dimensional CPU array interop through `ndarray`.
pub mod ndarray;

#[cfg(feature = "candle")]
/// CPU tensor interop through `candle-core`.
pub mod candle;