Struct tract_core::model::Node

pub struct Node<F: Fact + Hash, O: Hash> {
    pub id: usize,
    pub name: String,
    pub inputs: Vec<OutletId>,
    pub op: O,
    pub outputs: TVec<Outlet<F>>,
}

A Node in an Model.

Parameterized by a Fact implementation matching the one used in the model.

Fields

id: usize

node id in the model

Caution: this id will not be persistent during networks transformation

name: String

name of the node

This will usually come from the importing framework. tract transformation try to maintain the names accross transformations.

inputs: Vec<OutletId>

A list of incoming tensors, identified by the node outlet that creates them.

op: O

The actual operation the node performs.

outputs: TVec<Outlet<F>>

List of ouputs, with their descendant and tensor type information.

Implementations

impl<F, NodeOp> Node<F, NodeOp>where
    F: Fact + Hash,
    NodeOp: Debug + Display + AsRef<dyn Op> + AsMut<dyn Op> + Hash,

pub fn op(&self) -> &dyn Op

Access the op of the node

pub fn op_as<O: Op>(&self) -> Option<&O>

Try to downcast the node operation to O.

pub fn op_as_mut<O: Op>(&mut self) -> Option<&mut O>

Try to downcast the node operation to O.

pub fn op_is<O: Op>(&self) -> bool

Check if the node operation is of type O.

pub fn same_as(&self, other: &Node<F, NodeOp>) -> bool

Check that this node produce the same outputs as other.

Trait Implementations

impl<F: Clone + Fact + Hash, O: Clone + Hash> Clone for Node<F, O>

fn clone(&self) -> Node<F, O>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<F: Debug + Fact + Hash, O: Debug + Hash> Debug for Node<F, O>

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

Formats the value using the given formatter.

impl<F: Fact + Hash, O: Hash + Display> Display for Node<F, O>

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

Formats the value using the given formatter.

impl<F: Fact + Hash, O: Hash> Hash for Node<F, O>

fn hash<H: Hasher>(&self, state: &mut H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)where
    H: Hasher,
    Self: Sized,

Feeds a slice of this type into the given Hasher.