Enum SemiringCoreDescriptor 

pub enum SemiringCoreDescriptor {
    BatchedGemm {
        batch_dims: Vec<usize>,
        m: usize,
        n: usize,
        k: usize,
    },
    ReduceAdd {
        modes_a: Vec<u32>,
        modes_c: Vec<u32>,
    },
    Trace {
        modes_a: Vec<u32>,
        modes_c: Vec<u32>,
        paired: Vec<(u32, u32)>,
    },
    AntiTrace {
        modes_a: Vec<u32>,
        modes_c: Vec<u32>,
        paired: Vec<(u32, u32)>,
    },
    AntiDiag {
        modes_a: Vec<u32>,
        modes_c: Vec<u32>,
        paired: Vec<(u32, u32)>,
    },
    MakeContiguous,
}

Descriptor for semiring-core execution operations.

This is the minimal protocol family that tenferro-einsum may depend on.

Examples

use tenferro_prims::SemiringCoreDescriptor;

let desc = SemiringCoreDescriptor::MakeContiguous;
assert!(matches!(desc, SemiringCoreDescriptor::MakeContiguous));

Variants

BatchedGemm

Batched semiring GEMM.

Fields

batch_dims: Vec<usize>

Batch dimension sizes.

m: usize

Rows in A / C.

n: usize

Columns in B / C.

k: usize

Contracted dimension.

ReduceAdd

Reduction using semiring addition.

Fields

modes_a: Vec<u32>

Mode labels for input tensor A.

modes_c: Vec<u32>

Mode labels for output tensor C.

Trace

Diagonal contraction.

Fields

modes_a: Vec<u32>

Mode labels for input tensor A.

modes_c: Vec<u32>

Mode labels for output tensor C.

paired: Vec<(u32, u32)>

Paired diagonal modes.

AntiTrace

Diagonal scatter-add for trace adjoints.

Fields

modes_a: Vec<u32>

Mode labels for input tensor A.

modes_c: Vec<u32>

Mode labels for output tensor C.

paired: Vec<(u32, u32)>

Paired diagonal modes.

AntiDiag

Diagonal scatter/write for diag adjoints.

Fields

modes_a: Vec<u32>

Mode labels for input tensor A.

modes_c: Vec<u32>

Mode labels for output tensor C.

paired: Vec<(u32, u32)>

Paired diagonal modes.

MakeContiguous

Materialize a contiguous output tensor.

Trait Implementations

impl Clone for SemiringCoreDescriptor

fn clone(&self) -> SemiringCoreDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SemiringCoreDescriptor

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

Formats the value using the given formatter.

impl Hash for SemiringCoreDescriptor

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for SemiringCoreDescriptor

fn eq(&self, other: &SemiringCoreDescriptor) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for SemiringCoreDescriptor

impl StructuralPartialEq for SemiringCoreDescriptor