Enum TensorStorage 

pub enum TensorStorage {
    Owned(Arc<Vec<u8>>),
    View {
        data: *const u8,
        len: usize,
    },
}

Owned or borrowed tensor data

Variants

Owned(Arc<Vec<u8>>)

Owned data on CPU

View

View into external data (e.g., memory-mapped file)

Fields

data: *const u8

len: usize

Implementations

impl TensorStorage

pub fn owned(data: Vec<u8>) -> Self

pub unsafe fn view(data: *const u8, len: usize) -> Self

Safety

The data pointer must be valid for the lifetime of this storage.

pub fn as_bytes(&self) -> &[u8]

pub fn as_bytes_mut(&mut self) -> Option<&mut [u8]>

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn to_owned(&self) -> Self

Trait Implementations

impl Clone for TensorStorage

fn clone(&self) -> TensorStorage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorStorage

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

Formats the value using the given formatter.

impl Send for TensorStorage

impl Sync for TensorStorage