Enum CpuStorage 

pub enum CpuStorage {
    BF16(Vec<bf16>),
    F16(Vec<f16>),
    F32(Vec<f32>),
    I32(Vec<i32>),
    I64(Vec<i64>),
}

Variants

BF16(Vec<bf16>)

F16(Vec<f16>)

F32(Vec<f32>)

I32(Vec<i32>)

I64(Vec<i64>)

Implementations

impl CpuStorage

pub fn as_mut_ptr(&mut self) -> *mut c_void

pub fn as_ptr(&mut self) -> *const c_void

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn dtype(&self) -> DType

pub fn as_ref(&self) -> CpuStorageRef<'_>

pub fn as_mut_ref(&mut self) -> CpuStorageRefMut<'_>

pub fn data<T>(&self) -> Result<&[T], Error>

where T: WithDType,

pub fn data_mut<T>(&mut self) -> Result<&mut [T], Error>

where T: WithDType,

Trait Implementations

impl Clone for CpuStorage

fn clone(&self) -> CpuStorage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CpuStorage

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

Formats the value using the given formatter.

impl From<Vec<bf16>> for CpuStorage

fn from(value: Vec<bf16>) -> CpuStorage

Converts to this type from the input type.

impl From<Vec<f16>> for CpuStorage

fn from(value: Vec<f16>) -> CpuStorage

Converts to this type from the input type.

impl From<Vec<f32>> for CpuStorage

fn from(value: Vec<f32>) -> CpuStorage

Converts to this type from the input type.

impl From<Vec<i32>> for CpuStorage

fn from(value: Vec<i32>) -> CpuStorage

Converts to this type from the input type.

impl From<Vec<i64>> for CpuStorage

fn from(value: Vec<i64>) -> CpuStorage

Converts to this type from the input type.

impl Slice for CpuStorage

type Device = CpuDevice

fn len(&self) -> usize

fn dtype(&self) -> DType

fn device(&self) -> &<CpuStorage as Slice>::Device

fn copy_host_to_device<DT>(&mut self, src: &[DT]) -> Result<(), Error>

where DT: WithDType,

fn copy_device_to_host<DT>(&self, dst: &mut [DT]) -> Result<(), Error>

where DT: WithDType,

fn is_empty(&self) -> bool

fn to_vec<DT>(&self) -> Result<Vec<DT>, Error>

where DT: WithDType,