Enum TropicalPlan 

pub enum TropicalPlan<T: ScalarBase> {
    BatchedGemm {
        m: usize,
        n: usize,
        k: usize,
        _marker: PhantomData<T>,
    },
    Reduce {
        axis: usize,
        op: ReduceOp,
        _marker: PhantomData<T>,
    },
    Trace {
        paired: Vec<(u32, u32)>,
        _marker: PhantomData<T>,
    },
    Permute {
        perm: Vec<usize>,
        _marker: PhantomData<T>,
    },
    AntiTrace {
        paired: Vec<(u32, u32)>,
        _marker: PhantomData<T>,
    },
    AntiDiag {
        paired: Vec<(u32, u32)>,
        _marker: PhantomData<T>,
    },
}

Execution plan for tropical primitive operations on CPU.

Analogous to CpuPlan but for tropical algebras. The plan captures pre-computed kernel selection information.

Examples

use tenferro_prims::{CpuBackend, TensorPrims, PrimDescriptor, ReduceOp};
use tenferro_tropical::{MaxPlusAlgebra, TropicalPlan};

let desc = PrimDescriptor::Reduce {
    modes_a: vec![0, 1],
    modes_c: vec![0],
    op: ReduceOp::Max,
};
let plan: TropicalPlan<f64> = CpuBackend::plan::<f64>(&desc, &[&[3, 4], &[3]]).unwrap();

Variants

BatchedGemm

Plan for batched GEMM under tropical algebra.

Fields

m: usize

Number of rows.

n: usize

Number of columns.

k: usize

Contraction dimension.

_marker: PhantomData<T>

Reduce

Plan for reduction under tropical algebra.

Fields

axis: usize

Axis to reduce over.

op: ReduceOp

Reduction operation.

_marker: PhantomData<T>

Trace

Plan for trace under tropical algebra.

Fields

paired: Vec<(u32, u32)>

Paired modes.

_marker: PhantomData<T>

Permute

Plan for permutation.

Fields

perm: Vec<usize>

Permutation mapping.

_marker: PhantomData<T>

AntiTrace

Plan for anti-trace (AD backward).

Fields

paired: Vec<(u32, u32)>

Paired modes.

_marker: PhantomData<T>

AntiDiag

Plan for anti-diag (AD backward).

Fields

paired: Vec<(u32, u32)>

Paired modes.

_marker: PhantomData<T>