Skip to content

Graph Optimize

Graph optimization enums for configuring pattern handlers in NetsPresso's graph optimizer.

GraphOptimizePatternHandler

netspresso.enums.graph_optimize.GraphOptimizePatternHandler

Bases: StrEnum

Available Pattern Handlers

Category Pattern Name Status
Fuse PatternHandlerFoldingGeLU ✓
Fuse PatternHandlerFoldingSilu ✓
Fuse PatternHandlerFoldingRMSNorm ✓
Fuse PatternHandlerFoldingLayerNorm ✓
Optimize PatternHandlerChangeAxisOfSoftmax ✓
Replace PatternHandlerReplaceNegToConv ✓
Fuse PatternHandlerFuseBNToConv ✓
Fuse PatternHandlerFuseContinuousConcat ✓
Fuse PatternHandlerFuseMathIntoConv ✓
Replace PatternHandlerReplaceMatmulToConv ✓
Fuse PatternHandlerFuseMultiReshapeTranspose ✓
Fuse PatternHandlerFuseMultiBranchReshapeTranspose ✓
Remove PatternHandlerRemoveUselessSlice ✓

Usage Examples

Get All Available Pattern Handlers

from netspresso.enums.graph_optimize import GraphOptimizePatternHandler

# Get all available pattern handlers
all_patterns = GraphOptimizePatternHandler.get_all()
print(f"Available patterns: {len(all_patterns)}")