Struct ComputationGraph 

pub struct ComputationGraph {
    pub nodes: Vec<GraphNode>,
    pub edges: Vec<GraphEdge>,
    pub metadata: HashMap<String, String>,
}

Simplified computation graph representation for optimization

Fields

nodes: Vec<GraphNode>

Graph nodes (operations)

edges: Vec<GraphEdge>

Graph edges (data dependencies)

metadata: HashMap<String, String>

Graph metadata

Implementations

impl ComputationGraph

pub fn new() -> Self

Create a new empty computation graph

pub fn add_node(&mut self, node: GraphNode) -> usize

Add a node to the graph

pub fn add_edge(&mut self, edge: GraphEdge)

Add an edge to the graph

pub fn get_node(&self, id: usize) -> Option<&GraphNode>

Get node by ID

pub fn get_node_mut(&mut self, id: usize) -> Option<&mut GraphNode>

Get mutable node by ID

pub fn get_node_edges(&self, node_id: usize) -> Vec<&GraphEdge>

Get all edges connected to a node

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

Validate the graph structure

pub fn total_compute_cost(&self) -> f64

Calculate total estimated compute cost

pub fn total_memory_cost(&self) -> f64

Calculate total estimated memory cost

Trait Implementations

impl Clone for ComputationGraph

fn clone(&self) -> ComputationGraph

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ComputationGraph

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

Formats the value using the given formatter.

impl Default for ComputationGraph

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ComputationGraph

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for ComputationGraph

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

where __S: Serializer,

Serialize this value into the given Serde serializer.