aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/nn/rewrite/library/fc_sparsity24_layer.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/nn/rewrite/library/fc_sparsity24_layer.py')
-rw-r--r--src/mlia/nn/rewrite/library/fc_sparsity24_layer.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/src/mlia/nn/rewrite/library/fc_sparsity24_layer.py b/src/mlia/nn/rewrite/library/fc_sparsity24_layer.py
new file mode 100644
index 0000000..fda619f
--- /dev/null
+++ b/src/mlia/nn/rewrite/library/fc_sparsity24_layer.py
@@ -0,0 +1,16 @@
+# SPDX-FileCopyrightText: Copyright 2023, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""PLACEHOLDER for example rewrite with one fully connected 2:4 sparsity layer."""
+from typing import Any
+
+from keras.api._v2 import keras # Temporary workaround for now: MLIA-1107
+
+from .fc_layer import get_keras_model
+
+# get_keras_model(input_shape,output_shape)
+
+
+def get_keras_model24(input_shape: Any, output_shape: Any) -> keras.Model:
+ """Generate TensorFlow Lite model for rewrite."""
+ model = get_keras_model(input_shape, output_shape)
+ return model