Enum GraphOp 

pub enum GraphOp {
    MatMul,
    Conv2d,
    Add,
    Sub,
    Mul,
    Div,
    Relu,
    Sigmoid,
    Softmax,
    BatchNorm,
    LayerNorm,
    MaxPool2d,
    AvgPool2d,
    Reshape,
    Transpose,
    Concat {
        axis: i32,
    },
    Custom {
        op_type: String,
    },
    FusedConv2d {
        activation: ActivationFunction,
    },
    Gemm {
        alpha: i32,
        beta: i32,
    },
    FusedElementWise {
        base_op: Box<GraphOp>,
        activation: ActivationFunction,
    },
}

The operation type for a graph node.

Variants

MatMul

Matrix multiplication.

Conv2d

Convolution 2D.

Add

Element-wise addition.

Sub

Element-wise subtraction.

Mul

Element-wise multiplication.

Div

Element-wise division.

Relu

Rectified Linear Unit activation.

Sigmoid

Sigmoid activation.

Softmax

Softmax activation.

BatchNorm

Batch normalization.

LayerNorm

Layer normalization.

MaxPool2d

Max pooling 2D.

AvgPool2d

Average pooling 2D.

Reshape

Reshape/view.

Transpose

Transpose/permute dimensions.

Concat

Concatenation along an axis.

Fields

axis: i32

Custom

Custom/vendor-specific operation.

Fields

op_type: String

FusedConv2d

Fused Conv2D with optional bias addition and activation. Inputs: [input, weight, bias?]

Fields

activation: ActivationFunction

Gemm

General matrix multiply: alpha * A @ B + beta * C (ONNX Gemm). Inputs: [A, B, C]

Fields

alpha: i32

Scaling factor for the matrix product.

beta: i32

Scaling factor for the bias term.

FusedElementWise

Element-wise binary operation with fused activation. Inputs: [lhs, rhs]

Fields

base_op: Box<GraphOp>

The base element-wise operation (Add, Sub, Mul, Div).

activation: ActivationFunction

Implementations

impl GraphOp

pub fn onnx_op_type(&self) -> &str

Returns the ONNX operator type string.

Trait Implementations

impl Clone for GraphOp

fn clone(&self) -> GraphOp

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphOp

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

Formats the value using the given formatter.

impl Eq for GraphOp

impl PartialEq for GraphOp

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for GraphOp