Enum burn::backend::libtorch::Storage

pub enum Storage {
    View {
        buffer_ref: Arc<*mut c_void>,
        view_ref: Arc<*mut c_void>,
    },
    Owned {
        buffer_ref: Arc<*mut c_void>,
    },
}

A reference to a tensor storage.

Variants

View

When a tensor is a partial view of another tensor.

Fields

buffer_ref: Arc<*mut c_void>

Storage reference for the whole buffer.

view_ref: Arc<*mut c_void>

Storage reference for the partial buffer.

Owned

When a tensor use all of its buffer.

Fields

buffer_ref: Arc<*mut c_void>

Storage reference for the whole buffer.

Implementations

impl Storage

pub fn can_mut(&self) -> bool

Check if the storage can be used inplace.

pub fn buffer_ref(&self) -> &Arc<*mut c_void>

Get the whole buffer reference.

Trait Implementations

impl Clone for Storage

fn clone(&self) -> Storage

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Storage

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

Formats the value using the given formatter.

impl PartialEq for Storage

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for Storage