Struct CpuBackend 

pub struct CpuBackend;

Re-export CPU backend. CPU backend. Implements Backend by running operations on CPU via iterators.

Trait Implementations

impl Backend for CpuBackend

type Device = CpuDevice

The device type for this backend.

type Storage = CpuStorage

The storage type for this backend.

fn zeros( shape: &Shape, dtype: DType, _device: &CpuDevice, ) -> Result<CpuStorage, Error>

Allocate storage filled with zeros.

fn ones( shape: &Shape, dtype: DType, _device: &CpuDevice, ) -> Result<CpuStorage, Error>

Allocate storage filled with ones.

fn full( shape: &Shape, val: f64, dtype: DType, _device: &CpuDevice, ) -> Result<CpuStorage, Error>

Allocate storage filled with a constant value.

fn from_f64_slice( data: &[f64], dtype: DType, _device: &CpuDevice, ) -> Result<CpuStorage, Error>

Create storage from a flat f64 slice, converting to the target dtype.

fn rand_uniform( shape: &Shape, dtype: DType, _device: &CpuDevice, ) -> Result<CpuStorage, Error>

Create storage with random uniform values in [0, 1).

fn rand_normal( shape: &Shape, dtype: DType, _device: &CpuDevice, ) -> Result<CpuStorage, Error>

Create storage with random normal values (mean=0, std=1).

fn binary_op( op: BinaryOp, lhs: &CpuStorage, lhs_layout: &Layout, rhs: &CpuStorage, rhs_layout: &Layout, ) -> Result<CpuStorage, Error>

Apply a binary op element-wise: result[i] = op(lhs[i], rhs[i]). The layouts handle broadcasting and non-contiguous access.

fn unary_op( op: UnaryOp, input: &CpuStorage, layout: &Layout, ) -> Result<CpuStorage, Error>

Apply a unary op element-wise: result[i] = op(input[i]).

fn reduce_op( op: ReduceOp, input: &CpuStorage, layout: &Layout, dims: &[usize], keep_dim: bool, ) -> Result<CpuStorage, Error>

Reduce along specific dimensions. If dims is empty, reduce over all elements.

fn matmul( lhs: &CpuStorage, lhs_layout: &Layout, rhs: &CpuStorage, rhs_layout: &Layout, ) -> Result<CpuStorage, Error>

General matrix multiply: C = A @ B. Supports batched matmul for tensors with rank > 2.

fn to_contiguous( input: &CpuStorage, layout: &Layout, ) -> Result<CpuStorage, Error>

Make a contiguous copy of the storage following the given layout. If the layout is already contiguous, this may just clone the storage.

fn to_f64_vec(input: &CpuStorage, layout: &Layout) -> Result<Vec<f64>, Error>

Copy data from this storage to a Vec on the host (for inspection).

fn cmp_op( op: CmpOp, lhs: &CpuStorage, lhs_layout: &Layout, rhs: &CpuStorage, rhs_layout: &Layout, ) -> Result<CpuStorage, Error>

Element-wise comparison, returns a u8 storage (0 or 1).

fn affine( input: &CpuStorage, layout: &Layout, mul: f64, add: f64, ) -> Result<CpuStorage, Error>

Affine transform: result = input * mul + add. Used for normalization and other fused operations.

fn index_select( input: &CpuStorage, input_layout: &Layout, indices: &CpuStorage, indices_layout: &Layout, dim: usize, ) -> Result<CpuStorage, Error>

Gather elements along a dimension using index tensor.

fn powf( input: &CpuStorage, layout: &Layout, exponent: f64, ) -> Result<CpuStorage, Error>

Element-wise power: result[i] = input[i] ^ exponent.

fn clamp( input: &CpuStorage, layout: &Layout, min: f64, max: f64, ) -> Result<CpuStorage, Error>

Element-wise clamp: result[i] = clamp(input[i], min, max).

fn where_cond( mask: &CpuStorage, mask_layout: &Layout, on_true: &CpuStorage, on_true_layout: &Layout, on_false: &CpuStorage, on_false_layout: &Layout, ) -> Result<CpuStorage, Error>

Element-wise conditional: result[i] = if mask[i] != 0 { on_true[i] } else { on_false[i] }.

fn gather( input: &CpuStorage, input_layout: &Layout, index: &CpuStorage, index_layout: &Layout, dim: usize, ) -> Result<CpuStorage, Error>

Gather elements along dim using index tensor.

fn cat( inputs: &[(&CpuStorage, &Layout)], out_shape: &Shape, dim: usize, ) -> Result<CpuStorage, Error>

Concatenate multiple storages along dim into a single contiguous storage. Each entry is (storage, layout) so non-contiguous inputs are handled correctly. out_shape is the pre-validated output shape.

fn cast( input: &Self::Storage, layout: &Layout, dtype: DType, device: &Self::Device, ) -> Result<Self::Storage, Error>

Cast storage to a different dtype on-device (no host round-trip).

impl Clone for CpuBackend

fn clone(&self) -> CpuBackend

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CpuBackend

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

Formats the value using the given formatter.