Enum OnnxOpType 

pub enum OnnxOpType {
    Add,
    Sub,
    Mul,
    Div,
    MatMul,
    Gemm,
    Relu,
    Sigmoid,
    Tanh,
    Softmax,
    Gelu,
    Conv,
    ConvTranspose,
    MaxPool,
    AveragePool,
    GlobalAveragePool,
    BatchNormalization,
    LayerNormalization,
    Reshape,
    Transpose,
    Flatten,
    Squeeze,
    Unsqueeze,
    Concat,
    ReduceSum,
    ReduceMean,
    ReduceMax,
    ReduceMin,
    Dropout,
    Constant,
    Identity,
    Unknown,
}

ONNX operator type mapping.

Variants

Add

Element-wise addition.

Sub

Element-wise subtraction.

Mul

Element-wise multiplication.

Div

Element-wise division.

MatMul

Matrix multiplication.

Gemm

General matrix multiplication (with bias).

Relu

Rectified Linear Unit activation.

Sigmoid

Sigmoid activation.

Tanh

Hyperbolic tangent activation.

Softmax

Softmax activation.

Gelu

Gaussian Error Linear Unit activation.

Conv

Convolution operation.

ConvTranspose

Transposed convolution (deconvolution).

MaxPool

Max pooling operation.

AveragePool

Average pooling operation.

GlobalAveragePool

Global average pooling operation.

BatchNormalization

Batch normalization.

LayerNormalization

Layer normalization.

Reshape

Reshape tensor dimensions.

Transpose

Transpose tensor dimensions.

Flatten

Flatten tensor to 2D.

Squeeze

Remove dimensions of size 1.

Unsqueeze

Add dimension of size 1.

Concat

Concatenate tensors along axis.

ReduceSum

Sum reduction along axis.

ReduceMean

Mean reduction along axis.

ReduceMax

Max reduction along axis.

ReduceMin

Min reduction along axis.

Dropout

Dropout layer (training regularization).

Constant

Constant tensor.

Identity

Identity pass-through.

Unknown

Unknown or unsupported operator.

Implementations

impl OnnxOpType

pub fn from_str(s: &str) -> Self

Parse ONNX operator type from string.

pub fn as_str(&self) -> &'static str

Get the ONNX operator name.

Trait Implementations

impl Clone for OnnxOpType

fn clone(&self) -> OnnxOpType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OnnxOpType

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

Formats the value using the given formatter.

impl PartialEq for OnnxOpType

fn eq(&self, other: &OnnxOpType) -> 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 OnnxOpType

impl Eq for OnnxOpType

impl StructuralPartialEq for OnnxOpType