Enum CpuStorage

pub enum CpuStorage {
    U8(Vec<u8>),
    I8(Vec<i8>),
    U32(Vec<u32>),
    I16(Vec<i16>),
    I32(Vec<i32>),
    I64(Vec<i64>),
    BF16(Vec<bf16>),
    F16(Vec<f16>),
    F32(Vec<f32>),
    F64(Vec<f64>),
    F8E4M3(Vec<F8E4M3>),
}

Variants

U8(Vec<u8>)

I8(Vec<i8>)

U32(Vec<u32>)

I16(Vec<i16>)

I32(Vec<i32>)

I64(Vec<i64>)

BF16(Vec<bf16>)

F16(Vec<f16>)

F32(Vec<f32>)

F64(Vec<f64>)

F8E4M3(Vec<F8E4M3>)

Implementations

impl CpuStorage

pub fn as_slice<D: WithDType>(&self) -> Result<&[D]>

pub fn concat(storages: &[CpuStorage]) -> Result<CpuStorage>

Trait Implementations

impl BackendStorage for CpuStorage

type Device = CpuDevice

fn dtype(&self) -> DType

fn to_dtype(&self, layout: &Layout, dtype: DType) -> Result<Self>

fn reduce_op( &self, op: ReduceOp, layout: &Layout, reduce_dims: &[usize], ) -> Result<Self>

fn cmp( &self, op: CmpOp, rhs: &Self, lhs_l: &Layout, rhs_l: &Layout, ) -> Result<Self>

fn affine(&self, layout: &Layout, mul: f64, add: f64) -> Result<Self>

fn avg_pool2d( &self, layout: &Layout, kernel_size: (usize, usize), stride: (usize, usize), ) -> Result<Self>

fn max_pool2d( &self, layout: &Layout, kernel_size: (usize, usize), stride: (usize, usize), ) -> Result<Self>

fn upsample_nearest1d(&self, layout: &Layout, sz: usize) -> Result<Self>

fn upsample_nearest2d( &self, layout: &Layout, h: usize, w: usize, ) -> Result<Self>

fn powf(&self, layout: &Layout, e: f64) -> Result<Self>

fn elu(&self, layout: &Layout, alpha: f64) -> Result<Self>

fn unary_impl<B: UnaryOpT>(&self, layout: &Layout) -> Result<Self>

fn binary_impl<B: BinaryOpT>( &self, rhs: &Self, lhs_l: &Layout, rhs_l: &Layout, ) -> Result<Self>

fn copy2d( &self, dst: &mut Self, d1: usize, d2: usize, src_s: usize, dst_s: usize, src_o: usize, dst_o: usize, ) -> Result<()>

fn copy_strided_src( &self, dst: &mut Self, dst_offset: usize, src_l: &Layout, ) -> Result<()>

fn where_cond( &self, layout: &Layout, t: &Self, t_l: &Layout, f: &Self, f_l: &Layout, ) -> Result<Self>

fn conv1d( &self, l: &Layout, kernel: &Self, kernel_l: &Layout, params: &ParamsConv1D, ) -> Result<Self>

fn conv_transpose1d( &self, l: &Layout, kernel: &Self, kernel_l: &Layout, params: &ParamsConvTranspose1D, ) -> Result<Self>

fn conv2d( &self, l: &Layout, kernel: &Self, kernel_l: &Layout, params: &ParamsConv2D, ) -> Result<Self>

fn conv_transpose2d( &self, l: &Layout, kernel: &Self, kernel_l: &Layout, params: &ParamsConvTranspose2D, ) -> Result<Self>

fn index_select( &self, ids: &Self, l: &Layout, ids_l: &Layout, dim: usize, ) -> Result<Self>

fn gather( &self, l: &Layout, ids: &Self, ids_l: &Layout, dim: usize, ) -> Result<Self>

fn scatter_add( &self, l: &Layout, ids: &Self, ids_l: &Layout, src: &Self, src_l: &Layout, dim: usize, ) -> Result<Self>

fn index_add( &self, l: &Layout, ids: &Self, ids_l: &Layout, src: &Self, src_l: &Layout, dim: usize, ) -> Result<Self>

fn matmul_with_alpha_beta( &self, rhs: &Self, c: &mut Self, s: Option<f64>, bmnk: (usize, usize, usize, usize), lhs_l: &Layout, rhs_l: &Layout, c_l: &Layout, ) -> Result<()>

fn matmul_with_alpha( &self, rhs: &Self, s: Option<f64>, bmnk: (usize, usize, usize, usize), lhs_l: &Layout, rhs_l: &Layout, ) -> Result<Self>

fn device(&self) -> &Self::Device

fn try_clone(&self, _: &Layout) -> Result<Self>

fn to_cpu_storage(&self) -> Result<CpuStorage>

impl Clone for CpuStorage

fn clone(&self) -> CpuStorage

Returns a copy 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

Formats the value using the given formatter.