Struct tract_pulse::internal::tract_core::ops::nn::Node

pub struct Node<F, O>where
    F: Fact + Hash,
    O: Hash,{
    pub id: usize,
    pub name: String,
    pub inputs: Vec<OutletId, Global>,
    pub op: O,
    pub outputs: SmallVec<[Outlet<F>; 4]>,
}

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, Global>

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

op: O

The actual operation the node performs.

outputs: SmallVec<[Outlet<F>; 4]>

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 + 'static> + AsMut<dyn Op + 'static> + Hash,

pub fn op(&self) -> &(dyn Op + 'static)

Access the op of the node

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

Try to downcast the node operation to O.

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

Try to downcast the node operation to O.

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

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, O> Clone for Node<F, O>where
    F: Clone + Fact + Hash,
    O: Clone + Hash,

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, O> Debug for Node<F, O>where
    F: Debug + Fact + Hash,
    O: Debug + Hash,

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

Formats the value using the given formatter.

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

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

Formats the value using the given formatter.

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

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

Feeds this value into the given Hasher.

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

Feeds a slice of this type into the given Hasher.