Struct tvm::runtime::function::ffi::DLTensor

#[repr(C)]
pub struct DLTensor {
    pub data: *mut c_void,
    pub device: DLDevice,
    pub ndim: i32,
    pub dtype: DLDataType,
    pub shape: *mut i64,
    pub strides: *mut i64,
    pub byte_offset: u64,
}

\brief Plain C Tensor object, does not manage memory.

Fields

data: *mut c_void

\brief The opaque data pointer points to the allocated data. This will be CUDA device pointer or cl_mem handle in OpenCL. This pointer is always aligned to 256 bytes as in CUDA.

For given DLTensor, the size of memory required to store the contents of data is calculated as follows:

\code{.c} static inline size_t GetDataSize(const DLTensor* t) { size_t size = 1; for (tvm_index_t i = 0; i < t->ndim; ++i) { size *= t->shape[i]; } size *= (t->dtype.bits * t->dtype.lanes + 7) / 8; return size; } \endcode

device: DLDevice

\brief The device of the tensor

ndim: i32

\brief Number of dimensions

dtype: DLDataType

\brief The data type of the pointer

shape: *mut i64

\brief The shape of the tensor

strides: *mut i64

\brief strides of the tensor (in number of elements, not bytes) can be NULL, indicating tensor is compact and row-majored.

byte_offset: u64

\brief The offset in bytes to the beginning pointer to data

Trait Implementations

impl Clone for DLTensor

pub fn clone(&self) -> DLTensor

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DLTensor

pub fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for DLTensor

pub fn default() -> DLTensor

Returns the “default value” for a type.

impl<'a> From<&'a DLTensor> for ArgValue<'a>

pub fn from(arr: &'a DLTensor) -> ArgValue<'a>

Performs the conversion.

impl<'a, D> From<&'a mut ArrayBase<OwnedRepr<f32>, D>> for DLTensor where
    D: Dimension, 

pub fn from(arr: &'a mut ArrayBase<OwnedRepr<f32>, D>) -> DLTensor

Performs the conversion.

impl<'a, D> From<&'a mut ArrayBase<OwnedRepr<f64>, D>> for DLTensor where
    D: Dimension, 

pub fn from(arr: &'a mut ArrayBase<OwnedRepr<f64>, D>) -> DLTensor

Performs the conversion.

impl<'a, D> From<&'a mut ArrayBase<OwnedRepr<i32>, D>> for DLTensor where
    D: Dimension, 

pub fn from(arr: &'a mut ArrayBase<OwnedRepr<i32>, D>) -> DLTensor

Performs the conversion.

impl<'a, D> From<&'a mut ArrayBase<OwnedRepr<i64>, D>> for DLTensor where
    D: Dimension, 

pub fn from(arr: &'a mut ArrayBase<OwnedRepr<i64>, D>) -> DLTensor

Performs the conversion.

impl<'a, D> From<&'a mut ArrayBase<OwnedRepr<u32>, D>> for DLTensor where
    D: Dimension, 

pub fn from(arr: &'a mut ArrayBase<OwnedRepr<u32>, D>) -> DLTensor

Performs the conversion.

impl<'a, D> From<&'a mut ArrayBase<OwnedRepr<u64>, D>> for DLTensor where
    D: Dimension, 

pub fn from(arr: &'a mut ArrayBase<OwnedRepr<u64>, D>) -> DLTensor

Performs the conversion.

impl<'a> From<&'a mut DLTensor> for ArgValue<'a>

pub fn from(arr: &'a mut DLTensor) -> ArgValue<'a>

Performs the conversion.

impl PartialEq<DLTensor> for DLTensor

pub fn eq(&self, other: &DLTensor) -> bool

This method tests for self and other values to be equal, and is used by ==.

pub fn ne(&self, other: &DLTensor) -> bool

This method tests for !=.

impl Copy for DLTensor

impl Eq for DLTensor

impl StructuralEq for DLTensor

impl StructuralPartialEq for DLTensor