Enum FusedPattern 

pub enum FusedPattern {
    Single(KernelPattern),
    ConvBatchNorm {
        conv: KernelPattern,
        norm: Box<KernelPattern>,
    },
    WithActivation {
        base: Box<FusedPattern>,
        activation: FusedActivation,
        activation_pattern: Box<KernelPattern>,
    },
    MatMulBias {
        matmul: KernelPattern,
        bias_add: Box<KernelPattern>,
    },
}

A pattern that may be fused from one or more classified patterns.

Variants

Single(KernelPattern)

A single unfused pattern.

ConvBatchNorm

Conv2D followed by BatchNormalization.

Fields

conv: KernelPattern

norm: Box<KernelPattern>

WithActivation

A base pattern followed by an activation function.

Fields

base: Box<FusedPattern>

activation: FusedActivation

activation_pattern: Box<KernelPattern>

The original activation pattern (preserved for tensor connectivity).

MatMulBias

MatMul followed by Add (bias) — maps to ONNX Gemm.

Fields

matmul: KernelPattern

bias_add: Box<KernelPattern>

Implementations

impl FusedPattern

pub fn primary_pattern(&self) -> &KernelPattern

Return a reference to the primary pattern (for lowering).

pub fn fused_activation(&self) -> FusedActivation

Return the fused activation, if any.

Trait Implementations

impl Clone for FusedPattern

fn clone(&self) -> FusedPattern

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FusedPattern

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

Formats the value using the given formatter.

impl Display for FusedPattern

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

Formats the value using the given formatter.