Enum EagerBackend 

pub enum EagerBackend {
    Cpu(CpuBackend),
}

Variants

Cpu(CpuBackend)

Trait Implementations

impl BackendCachedDot for EagerBackend

fn dot_general_read_into_accum_cached( &mut self, _cache: &mut Self::RuntimeCache, _cache_slot: Option<usize>, lhs: TensorRead<'_>, rhs: TensorRead<'_>, config: &DotGeneralConfig, accumulation: DotGeneralAccumulation, out: TensorWrite<'_>, ) -> Result<(), Error>

Apply cached scaled dot-general accumulation into caller-provided output.

impl BackendSessionHost for EagerBackend

fn with_backend_session<R: Send>( &mut self, f: impl FnOnce(&mut dyn BackendSession) -> R + Send, ) -> R

impl Debug for EagerBackend

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

Formats the value using the given formatter.

impl TensorAnalytic for EagerBackend

fn exp(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn exp_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn log(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn log_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn sin(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn sin_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn cos(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn cos_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn tanh(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn tanh_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn sqrt(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn sqrt_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn rsqrt(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn rsqrt_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn pow(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn pow_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

fn expm1(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn expm1_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn log1p(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn log1p_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

impl TensorBuffer for EagerBackend

fn reclaim_buffer(&mut self, tensor: Tensor)

impl TensorDeviceTransfer for EagerBackend

fn download_to_host(&mut self, tensor: &Tensor) -> TensorResult<Tensor>

fn upload_host_tensor(&mut self, tensor: &Tensor) -> TensorResult<Tensor>

impl TensorDot for EagerBackend

fn dot_general( &mut self, lhs: &Tensor, rhs: &Tensor, config: &DotGeneralConfig, ) -> TensorResult<Tensor>

fn dot_general_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, config: &DotGeneralConfig, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with dot-general from read inputs.

fn dot_general_read_into_accum( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, config: &DotGeneralConfig, accumulation: DotGeneralAccumulation, out: TensorWrite<'_>, ) -> Result<(), Error>

Apply scaled dot-general accumulation into caller-provided output.

impl TensorElementwise for EagerBackend

fn add(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn add_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

Elementwise addition accepting either owned tensors or borrowed views.

fn sub(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn sub_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

Elementwise subtraction accepting either owned tensors or borrowed views.

fn mul(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn mul_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

fn neg(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn neg_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn conj(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn conj_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn div(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn div_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

fn rem(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

Elementwise remainder.

fn rem_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

Elementwise remainder accepting owned tensors or borrowed views.

fn abs(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn abs_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn sign(&mut self, input: &Tensor) -> TensorResult<Tensor>

fn sign_read(&mut self, input: TensorRead<'_>) -> TensorResult<Tensor>

fn maximum(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn maximum_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

fn minimum(&mut self, lhs: &Tensor, rhs: &Tensor) -> TensorResult<Tensor>

fn minimum_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, ) -> TensorResult<Tensor>

fn compare( &mut self, lhs: &Tensor, rhs: &Tensor, dir: &CompareDir, ) -> TensorResult<Tensor>

fn compare_read( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, dir: &CompareDir, ) -> TensorResult<Tensor>

fn select( &mut self, pred: &Tensor, on_true: &Tensor, on_false: &Tensor, ) -> TensorResult<Tensor>

fn select_read( &mut self, pred: TensorRead<'_>, on_true: TensorRead<'_>, on_false: TensorRead<'_>, ) -> TensorResult<Tensor>

fn clamp( &mut self, input: &Tensor, lower: &Tensor, upper: &Tensor, ) -> TensorResult<Tensor>

fn clamp_read( &mut self, input: TensorRead<'_>, lower: TensorRead<'_>, upper: TensorRead<'_>, ) -> TensorResult<Tensor>

fn add_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise addition.

fn add_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise addition from reads.

fn sub_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise subtraction.

fn sub_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise subtraction from reads.

fn mul_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise multiplication.

fn mul_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise multiplication from reads.

fn neg_into( &mut self, input: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise negation.

fn neg_read_into( &mut self, input: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise negation from a read.

fn conj_into( &mut self, input: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise conjugation.

fn conj_read_into( &mut self, input: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise conjugation from a read.

fn div_into( &mut self, lhs: &Tensor, rhs: &Tensor, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise division.

fn div_read_into( &mut self, lhs: TensorRead<'_>, rhs: TensorRead<'_>, out: TensorWrite<'_>, ) -> Result<(), Error>

Overwrite caller-provided output with elementwise division from reads.

impl TensorIndexing for EagerBackend

fn gather( &mut self, operand: &Tensor, start_indices: &Tensor, config: &GatherConfig, ) -> TensorResult<Tensor>

fn scatter( &mut self, operand: &Tensor, scatter_indices: &Tensor, updates: &Tensor, config: &ScatterConfig, ) -> TensorResult<Tensor>

fn slice( &mut self, input: &Tensor, config: &SliceConfig, ) -> TensorResult<Tensor>

fn dynamic_slice( &mut self, input: &Tensor, starts: &Tensor, slice_sizes: &[usize], ) -> TensorResult<Tensor>

fn dynamic_update_slice( &mut self, operand: &Tensor, update: &Tensor, starts: &Tensor, ) -> TensorResult<Tensor>

fn pad(&mut self, input: &Tensor, config: &PadConfig) -> TensorResult<Tensor>

fn concatenate( &mut self, inputs: &[&Tensor], axis: usize, ) -> TensorResult<Tensor>

fn reverse(&mut self, input: &Tensor, axes: &[usize]) -> TensorResult<Tensor>

impl TensorReduction for EagerBackend

fn reduce_sum(&mut self, input: &Tensor, axes: &[usize]) -> TensorResult<Tensor>

fn reduce_prod( &mut self, input: &Tensor, axes: &[usize], ) -> TensorResult<Tensor>

fn reduce_max(&mut self, input: &Tensor, axes: &[usize]) -> TensorResult<Tensor>

fn reduce_min(&mut self, input: &Tensor, axes: &[usize]) -> TensorResult<Tensor>

fn reduce_sum_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Sum elements across axes from an owned tensor or borrowed view.

fn reduce_prod_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Multiply elements across axes from an owned tensor or borrowed view.

fn reduce_max_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Take maximum values across axes from an owned tensor or borrowed view.

fn reduce_min_read( &mut self, input: TensorRead<'_>, axes: &[usize], ) -> Result<Tensor, Error>

Take minimum values across axes from an owned tensor or borrowed view.

impl TensorStructural for EagerBackend

fn transpose(&mut self, input: &Tensor, perm: &[usize]) -> TensorResult<Tensor>

fn reshape(&mut self, input: &Tensor, shape: &[usize]) -> TensorResult<Tensor>

fn reshape_read( &mut self, input: TensorRead<'_>, shape: &[usize], ) -> TensorResult<Tensor>

fn broadcast_in_dim( &mut self, input: &Tensor, shape: &[usize], dims: &[usize], ) -> TensorResult<Tensor>

fn broadcast_in_dim_read( &mut self, input: TensorRead<'_>, shape: &[usize], dims: &[usize], ) -> TensorResult<Tensor>

fn cast(&mut self, input: &Tensor, to: DType) -> TensorResult<Tensor>

Cast a tensor to another dtype using explicit dtype projection.

fn extract_diagonal( &mut self, input: &Tensor, axis_a: usize, axis_b: usize, ) -> TensorResult<Tensor>

fn embed_diagonal( &mut self, input: &Tensor, axis_a: usize, axis_b: usize, ) -> TensorResult<Tensor>

fn tril(&mut self, input: &Tensor, k: i64) -> TensorResult<Tensor>

fn triu(&mut self, input: &Tensor, k: i64) -> TensorResult<Tensor>

fn transpose_read( &mut self, input: TensorRead<'_>, perm: &[usize], ) -> Result<Tensor, Error>

fn convert(&mut self, input: &Tensor, to: DType) -> Result<Tensor, Error>

Convert a tensor to another dtype using checked dtype conversion.

impl BackendRuntimeCache for EagerBackend

impl TensorBackend for EagerBackend

impl TensorFusion for EagerBackend