Struct Graph 

pub struct Graph {
    pub name: String,
    pub config: Option<ModelConfig>,
    pub nodes: Vec<Node>,
    pub weights: HashMap<String, TensorInfo>,
    /* private fields */
}

The computation graph — the central IR artifact.

Fields

name: String

config: Option<ModelConfig>

nodes: Vec<Node>

weights: HashMap<String, TensorInfo>

Maps weight names to their tensor info.

Implementations

impl Graph

pub fn new(name: impl Into<String>) -> Self

pub fn with_config(self, config: ModelConfig) -> Self

pub fn add_node( &mut self, op: Op, inputs: Vec<NodeId>, output: TensorInfo, ) -> NodeId

Add a node to the graph and return its ID.

pub fn alloc_tensor_id(&mut self) -> TensorId

Allocate a new tensor ID.

pub fn register_weight( &mut self, name: String, shape: Shape, dtype: DType, ) -> TensorId

Register a weight tensor in the graph.

pub fn load_weight( &mut self, name: impl Into<String>, shape: Shape, dtype: DType, ) -> NodeId

Add a weight-loading node.

pub fn input( &mut self, name: impl Into<String>, shape: Shape, dtype: DType, ) -> NodeId

Add an input node.

pub fn node(&self, id: NodeId) -> &Node

Get a node by ID.

pub fn output_info(&self, id: NodeId) -> &TensorInfo

Get the output tensor info for a node.

pub fn len(&self) -> usize

Number of nodes in the graph.

pub fn is_empty(&self) -> bool

Whether the graph has no nodes.

pub fn topological_order(&self) -> Vec<NodeId>

Return node IDs in topological order. Since nodes are added in dependency order, this is just 0..n.

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

Validate the graph: check that all input references are valid and precede the node that uses them.

Trait Implementations

impl Clone for Graph

fn clone(&self) -> Graph

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Graph

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Graph

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for Graph

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

where __S: Serializer,

Serialize this value into the given Serde serializer.