diff options
| author | Madeleine Dunn <madeleine.dunn@arm.com> | 2023-10-05 18:19:23 +0100 |
|---|---|---|
| committer | Madeleine Dunn <madeleine.dunn@arm.com> | 2024-04-03 16:06:42 +0100 |
| commit | a67bcd6eccaa4de00a08a3def8323debecff84d7 (patch) | |
| tree | 2fa3d79a02de4de1e937f21dabe5766073a4a067 /src/mlia/nn/rewrite/core/rewrite.py | |
| parent | 879138ebe9dd4b9387bddbdf4af45ba390f14bf6 (diff) | |
| download | mlia-a67bcd6eccaa4de00a08a3def8323debecff84d7.tar.gz | |
feat: Implement sparsity
- Add a placeholder file and registry option for sparsity
Resolves: MLIA-999
Signed-off-by: Madeleine Dunn <madeleine.dunn@arm.com>
Change-Id: I273192ba6813309f5226e3d8e0b686ce87ee6b79
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", + ), ] ) |