Struct LowRankInferencePass 

pub struct LowRankInferencePass { /* private fields */ }

Optimization pass that scans an EinsumGraph and annotates Einsum nodes with low-rank approximation candidates.

Currently uses a heuristic based on einsum spec complexity (number of unique contracted indices) to identify potential candidates.

Implementations

impl LowRankInferencePass

pub fn new(config: LowRankConfig) -> Self

Create a new pass with the given configuration.

pub fn find_candidates(&self, graph: &EinsumGraph) -> Vec<LowRankCandidate>

Scan the graph and return a list of low-rank candidates.

pub fn apply_annotations(&self, graph: &EinsumGraph) -> LowRankPassStats

Apply annotations and return aggregate stats.

In this implementation the “annotation” is a dry-run analysis only — the graph is not mutated (annotation requires mutable access and is outside the scope of a read-only pass).

Trait Implementations

impl Debug for LowRankInferencePass

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

Formats the value using the given formatter.