diff options
Diffstat (limited to 'src/mlia/nn/rewrite/core/rewrite.py')
| -rw-r--r-- | src/mlia/nn/rewrite/core/rewrite.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/mlia/nn/rewrite/core/rewrite.py b/src/mlia/nn/rewrite/core/rewrite.py index f5f5561..03af0e9 100644 --- a/src/mlia/nn/rewrite/core/rewrite.py +++ b/src/mlia/nn/rewrite/core/rewrite.py @@ -115,7 +115,11 @@ class RewritingOptimizer(Optimizer): [ DynamicallyLoadedRewrite( "fully-connected", "mlia.nn.rewrite.library.fc_layer.get_keras_model" - ) + ), + DynamicallyLoadedRewrite( + "fully_connected_sparsity24", + "mlia.nn.rewrite.library.fc_sparsity24_layer.get_keras_model24", + ), ] ) |