Struct Graph 

pub struct Graph { /* private fields */ }

Computation graph for JIT compilation.

Implementations

impl Graph

pub fn new() -> Self

Creates a new empty graph.

pub fn add_node(&mut self, op: Op, dtype: DataType, shape: Shape) -> NodeId

Adds a node to the graph.

pub fn register_input(&mut self, name: &str, id: NodeId)

Registers an input node.

pub fn register_output(&mut self, name: &str, id: NodeId)

Registers an output node.

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

Returns the node for an ID.

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

Returns mutable node for an ID.

pub fn nodes(&self) -> &[Node]

Returns all nodes.

pub fn len(&self) -> usize

Returns the number of nodes.

pub fn is_empty(&self) -> bool

Returns whether the graph is empty.

pub fn inputs(&self) -> &FxHashMap<String, NodeId>

Returns input names and node IDs.

pub fn outputs(&self) -> &FxHashMap<String, NodeId>

Returns output names and node IDs.

pub fn input(&self, name: &str) -> Option<NodeId>

Returns the input node ID for a name.

pub fn output(&self, name: &str) -> Option<NodeId>

Returns the output node ID for a name.

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

Returns nodes in topological order.

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

Validates the graph structure.

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 Default for Graph

fn default() -> Self

Returns the “default value” for a type.