Struct CpuStorage 

pub struct CpuStorage(pub CpuStorage);

CPU backend storage: owns raw tensor data as a typed HostBuffer.

Defined in fuel-cpu-backend so the orphan rule allows implementing DynBackendStorage here with full access to CPU kernels.

Tuple Fields

0: CpuStorage

Implementations

impl CpuStorage

pub fn into_inner(self) -> HostBuffer

Unwrap the inner HostBuffer.

pub fn inner(&self) -> &HostBuffer

Borrow the inner HostBuffer.

Trait Implementations

impl Clone for CpuStorage

fn clone(&self) -> Self

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

Formats the value using the given formatter.

impl DynBackendStorage for CpuStorage

fn try_clone_dyn(&self, _layout: &Layout) -> Result<Box<dyn DynBackendStorage>>

Clone the elements described by layout into a new storage.

fn dtype_dyn(&self) -> DType

The element dtype of this storage.

fn device_dyn(&self) -> &dyn DynBackendDevice

The device that owns this storage.

fn device_arc_dyn(&self) -> Arc<dyn DynBackendDevice>

Return a cloned Arc handle to the owning device.

fn to_host_buffer_dyn(&self) -> Result<HostBuffer>

Copy the entire storage to a HostBuffer.

fn affine_dyn( &self, layout: &Layout, mul: f64, add: f64, ) -> Result<Box<dyn DynBackendStorage>>

Elementwise affine: x * mul + add.

fn powf_dyn( &self, layout: &Layout, e: f64, ) -> Result<Box<dyn DynBackendStorage>>

Raise every element to the power e.

fn elu_dyn( &self, layout: &Layout, alpha: f64, ) -> Result<Box<dyn DynBackendStorage>>

ELU activation.

fn reduce_op_dyn( &self, op: ReduceOp, layout: &Layout, axes: &[usize], ) -> Result<Box<dyn DynBackendStorage>>

Reduction (sum, min, max, …) over axes.

fn cmp_dyn( &self, op: CmpOp, rhs: &dyn DynBackendStorage, lhs_layout: &Layout, rhs_layout: &Layout, ) -> Result<Box<dyn DynBackendStorage>>

Elementwise comparison, returning a U8 buffer.

fn to_dtype_dyn( &self, layout: &Layout, dtype: DType, ) -> Result<Box<dyn DynBackendStorage>>

Cast to a different dtype.

fn unary_op_dyn( &self, layout: &Layout, op: UnaryOp, ) -> Result<Box<dyn DynBackendStorage>>

Elementwise unary operation.

fn binary_op_dyn( &self, rhs: &dyn DynBackendStorage, lhs_layout: &Layout, rhs_layout: &Layout, op: BinaryOp, ) -> Result<Box<dyn DynBackendStorage>>

Elementwise binary operation.

fn where_cond_dyn( &self, cond_layout: &Layout, on_true: &dyn DynBackendStorage, on_true_layout: &Layout, on_false: &dyn DynBackendStorage, on_false_layout: &Layout, ) -> Result<Box<dyn DynBackendStorage>>

Select on_true where condition is nonzero, on_false elsewhere.

fn conv1d_dyn( &self, l: &Layout, kernel: &dyn DynBackendStorage, kernel_l: &Layout, params: &ParamsConv1D, ) -> Result<Box<dyn DynBackendStorage>>

1-D convolution.

fn conv_transpose1d_dyn( &self, l: &Layout, kernel: &dyn DynBackendStorage, kernel_l: &Layout, params: &ParamsConvTranspose1D, ) -> Result<Box<dyn DynBackendStorage>>

1-D transposed convolution.

fn conv2d_dyn( &self, l: &Layout, kernel: &dyn DynBackendStorage, kernel_l: &Layout, params: &ParamsConv2D, ) -> Result<Box<dyn DynBackendStorage>>

2-D convolution.

fn conv_transpose2d_dyn( &self, l: &Layout, kernel: &dyn DynBackendStorage, kernel_l: &Layout, params: &ParamsConvTranspose2D, ) -> Result<Box<dyn DynBackendStorage>>

2-D transposed convolution.

fn avg_pool2d_dyn( &self, layout: &Layout, kernel: (usize, usize), stride: (usize, usize), ) -> Result<Box<dyn DynBackendStorage>>

2-D average pooling.

fn max_pool2d_dyn( &self, layout: &Layout, kernel: (usize, usize), stride: (usize, usize), ) -> Result<Box<dyn DynBackendStorage>>

2-D max pooling.

fn upsample_nearest1d_dyn( &self, layout: &Layout, target_size: usize, ) -> Result<Box<dyn DynBackendStorage>>

1-D nearest-neighbor upsampling.

fn upsample_nearest2d_dyn( &self, layout: &Layout, target_h: usize, target_w: usize, ) -> Result<Box<dyn DynBackendStorage>>

2-D nearest-neighbor upsampling.

fn upsample_bilinear2d_dyn( &self, layout: &Layout, target_h: usize, target_w: usize, align_corners: bool, scale_h: Option<f64>, scale_w: Option<f64>, ) -> Result<Box<dyn DynBackendStorage>>

2-D bilinear upsampling.

fn gather_dyn( &self, src_layout: &Layout, ids: &dyn DynBackendStorage, ids_layout: &Layout, dim: usize, ) -> Result<Box<dyn DynBackendStorage>>

Gather elements along dim using ids.

fn scatter_set_dyn( &mut self, self_layout: &Layout, src: &dyn DynBackendStorage, src_layout: &Layout, ids: &dyn DynBackendStorage, ids_layout: &Layout, dim: usize, ) -> Result<()>

Scatter-set src into self along dim at positions in ids.

fn scatter_add_set_dyn( &mut self, self_layout: &Layout, src: &dyn DynBackendStorage, src_layout: &Layout, ids: &dyn DynBackendStorage, ids_layout: &Layout, dim: usize, ) -> Result<()>

Scatter-add src into self along dim at positions in ids.

fn index_select_dyn( &self, ids: &dyn DynBackendStorage, src_layout: &Layout, ids_layout: &Layout, dim: usize, ) -> Result<Box<dyn DynBackendStorage>>

Select slices along dim at positions in ids.

fn index_add_dyn( &self, self_layout: &Layout, ids: &dyn DynBackendStorage, ids_layout: &Layout, src: &dyn DynBackendStorage, src_layout: &Layout, dim: usize, ) -> Result<Box<dyn DynBackendStorage>>

Index-add: accumulate src slices into self along dim at ids.

fn matmul_dyn( &self, rhs: &dyn DynBackendStorage, bmnk: (usize, usize, usize, usize), lhs_layout: &Layout, rhs_layout: &Layout, ) -> Result<Box<dyn DynBackendStorage>>

Matrix multiply.

fn copy_strided_src_dyn( &self, dst: &mut dyn DynBackendStorage, dst_offset: usize, src_layout: &Layout, ) -> Result<()>

Copy strided source into dst starting at dst_offset.

fn copy2d_dyn( &self, dst: &mut dyn DynBackendStorage, d1: usize, d2: usize, src_stride1: usize, dst_stride1: usize, src_offset: usize, dst_offset: usize, ) -> Result<()>

Copy a 2-D tile from self into dst.

fn const_set_dyn(&mut self, value: Scalar, layout: &Layout) -> Result<()>

Set every element described by layout to value.

fn as_any(&self) -> &dyn Any

Downcast to the concrete storage type.

fn as_any_mut(&mut self) -> &mut dyn Any

Downcast to the concrete storage type (mutable).

fn to_cpu_storage_dyn(&self) -> Result<HostBuffer, Error>

Deprecated alias for to_host_buffer_dyn.

impl From<CpuStorage> for CpuStorage

fn from(s: CpuStorage) -> Self

Converts to this type from the input type.

impl From<HostBuffer> for CpuStorage

fn from(s: HostBuffer) -> Self

Converts to this type from the input type.

impl HostStorage for CpuStorage

fn as_host_buffer_ref(&self) -> Result<HostBufferRef<'_>>

Borrow the underlying data as a HostBufferRef (zero-copy).

fn into_host_buffer(self) -> Result<HostBuffer>

Extract the underlying data as an owned HostBuffer.