Struct EinsumGraph 

pub struct EinsumGraph {
    pub tensors: Vec<String>,
    pub nodes: Vec<EinsumNode>,
    pub inputs: Vec<usize>,
    pub outputs: Vec<usize>,
    pub tensor_metadata: HashMap<usize, Metadata>,
}

Fields

tensors: Vec<String>

nodes: Vec<EinsumNode>

inputs: Vec<usize>

outputs: Vec<usize>

tensor_metadata: HashMap<usize, Metadata>

Metadata for tensors (indexed by tensor index)

Implementations

impl EinsumGraph

pub fn extract_subgraph( &self, node_indices: &[usize], ) -> Result<EinsumGraph, IrError>

Extract a subgraph containing only the specified nodes and their dependencies.

pub fn merge( &mut self, other: &EinsumGraph, ) -> Result<HashMap<usize, usize>, IrError>

Merge another graph into this one.

Returns a mapping from old tensor indices to new tensor indices.

pub fn visit<V: GraphVisitor>(&self, visitor: &mut V)

Visit all nodes in the graph using a visitor.

pub fn visit_mut<V: GraphMutVisitor>( &mut self, visitor: &mut V, ) -> Result<(), IrError>

Visit all nodes mutably using a mutable visitor.

pub fn apply_rewrite<F>(&mut self, rule: F) -> Result<usize, IrError>

where F: FnMut(&EinsumNode) -> Option<EinsumNode>,

Apply a rewrite rule to all nodes in the graph.

The rule function takes a node and returns an optional replacement node.

pub fn tensor_consumers(&self, tensor_idx: usize) -> Vec<usize>

Get all nodes that depend on a specific tensor (consume it as input).

pub fn tensor_producer(&self, tensor_idx: usize) -> Option<usize>

Get the node that produces a specific tensor.

Note: In the current graph model, tensors can be produced by at most one node or be external inputs. This returns nodes that might output to this tensor based on graph topology.

pub fn has_path(&self, node_from: usize, node_to: usize) -> bool

Check if there’s a path from node_from to node_to based on node ordering.

pub fn dependencies(&self, node_idx: usize) -> HashSet<usize>

Get dependency chain for a node (all nodes it depends on).

pub fn node_count(&self) -> usize

Get number of nodes.

pub fn tensor_count(&self) -> usize

Get number of tensors.

impl EinsumGraph

pub fn new() -> Self

pub fn with_capacity(tensor_cap: usize, node_cap: usize) -> Self

pub fn add_tensor(&mut self, name: impl Into<String>) -> usize

pub fn add_node(&mut self, node: EinsumNode) -> Result<usize, IrError>

pub fn add_input(&mut self, tensor_idx: usize) -> Result<(), IrError>

pub fn add_output(&mut self, tensor_idx: usize) -> Result<(), IrError>

pub fn validate(&self) -> Result<(), IrError>

pub fn is_empty(&self) -> bool

pub fn add_tensor_metadata(&mut self, tensor_idx: usize, metadata: Metadata)

Add metadata for a tensor.

pub fn get_tensor_metadata(&self, tensor_idx: usize) -> Option<&Metadata>

Get metadata for a tensor if it exists.

pub fn add_tensor_with_metadata( &mut self, name: impl Into<String>, metadata: Metadata, ) -> usize

Add a tensor with metadata.

Trait Implementations

impl Clone for EinsumGraph

fn clone(&self) -> EinsumGraph

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for EinsumGraph

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

Formats the value using the given formatter.

impl Default for EinsumGraph

fn default() -> EinsumGraph

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for EinsumGraph

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for EinsumGraph

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

Formats the value using the given formatter.

impl PartialEq for EinsumGraph

fn eq(&self, other: &EinsumGraph) -> 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 Serialize for EinsumGraph

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for EinsumGraph