Enum NodeType 

pub enum NodeType {
    Linear,
    Conv,
    LayerNorm,
    BatchNorm,
    Embedding,
    Attention,
    MLP,
    Activation,
    Pooling,
    Input,
    Output,
    Other,
}

Type of layer/operation node.

Variants

Linear

Linear/Dense layer.

Conv

Convolutional layer.

LayerNorm

Layer normalization.

BatchNorm

Batch normalization.

Embedding

Embedding layer.

Attention

Attention block.

MLP

MLP/FFN block.

Activation

Activation function.

Pooling

Pooling layer.

Input

Input node.

Output

Output node.

Other

Other/unknown.

Trait Implementations

impl Clone for NodeType

fn clone(&self) -> NodeType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for NodeType

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

Formats the value using the given formatter.

impl PartialEq for NodeType

fn eq(&self, other: &NodeType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for NodeType

impl Eq for NodeType

impl StructuralPartialEq for NodeType