Struct GraphBuilder 

pub struct GraphBuilder { /* private fields */ }

Graph builder for convenient graph construction.

Implementations

impl GraphBuilder

pub fn new() -> Self

Create a new graph builder.

pub fn add_op(&mut self, op_type: &str, name: Option<String>) -> NodeId

Add a node with the given operation type.

pub fn add_input(&mut self, node_id: NodeId, tensor_name: &str) -> &mut Self

Add an input tensor to a node.

pub fn add_output(&mut self, node_id: NodeId, tensor_name: &str) -> &mut Self

Add an output tensor to a node.

pub fn add_attribute( &mut self, node_id: NodeId, name: &str, value: AttributeValue, ) -> &mut Self

Add an attribute to a node.

pub fn connect( &mut self, from_node: NodeId, to_node: NodeId, tensor_name: &str, ) -> Result<&mut Self>

Connect two nodes with a tensor.

pub fn set_inputs(&mut self, inputs: Vec<String>) -> &mut Self

Set graph inputs.

pub fn set_outputs(&mut self, outputs: Vec<String>) -> &mut Self

Set graph outputs.

pub fn build(self) -> Result<ModelGraph>

Build the final graph.

Trait Implementations

impl Default for GraphBuilder

fn default() -> Self

Returns the “default value” for a type.