Struct Node 

pub struct Node {
    pub id: NodeId,
    pub name: String,
    pub op_type: String,
    pub domain: String,
    pub inputs: Vec<Option<ValueId>>,
    pub outputs: Vec<ValueId>,
    pub attributes: HashMap<String, Attribute>,
    pub doc_string: Option<String>,
    pub device: Option<DeviceId>,
    pub exec_order: Option<usize>,
}

An operation in the graph.

Inputs are Option<ValueId> because ONNX ops may have optional (skipped) inputs represented by empty names; a None slot preserves positional arity. Outputs are always present (SSA values).

Fields

id: NodeId

name: String

Optional ONNX node name ("" means unnamed).

op_type: String

domain: String

Operator domain ("" == the default ONNX domain).

inputs: Vec<Option<ValueId>>

outputs: Vec<ValueId>

attributes: HashMap<String, Attribute>

doc_string: Option<String>

device: Option<DeviceId>

Device placement, filled in by the placement pass.

exec_order: Option<usize>

Position in the final execution schedule, filled in by the scheduler.

Implementations

impl Node

pub fn new( id: NodeId, op_type: impl Into<String>, inputs: Vec<Option<ValueId>>, outputs: Vec<ValueId>, ) -> Self

A new node with the given op type and edges, and no attributes.

Examples found in repository

examples/remove_node_bench.rs (lines 13-18)

fn hub_graph(node_count: usize) -> (Graph, Vec<NodeId>) {
    let mut graph = Graph::new();
    let hub = graph.create_value(DataType::Float32, static_shape([1]));
    graph.add_input(hub);
    let mut nodes = Vec::with_capacity(node_count);
    for _ in 0..node_count {
        let output = graph.create_value(DataType::Float32, static_shape([1]));
        nodes.push(graph.insert_node(Node::new(
            NodeId(0),
            "Identity",
            vec![Some(hub)],
            vec![output],
        )));
    }
    (graph, nodes)
}

pub fn input_values(&self) -> impl Iterator<Item = ValueId> + '_

Iterate over the present (non-skipped) input value ids.

pub fn attr(&self, name: &str) -> Option<&Attribute>

Look up an attribute by name.

Trait Implementations

impl Clone for Node

fn clone(&self) -> Node

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Node

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

Formats the value using the given formatter.