feat: CLI and API changes for the conv2d rewrites

Implements CLI and API changes for the new conv2d rewrite targets

Resolves: MLIA-1157

Signed-off-by: Nathan Bailey <nathan.bailey@arm.com>
Change-Id: I03c7a3a536d2f0a805b4689a9d96b95f8b4ab86c

1 files changed, 0 insertions, 26 deletions

diff --git a/src/mlia/nn/rewrite/library/fc_clustering_layer.py b/src/mlia/nn/rewrite/library/fc_clustering_layer.py
deleted file mode 100644
index 7cc383e..0000000
--- a/src/mlia/nn/rewrite/library/fc_clustering_layer.py
+++ /dev/null
@@ -1,26 +0,0 @@
-# SPDX-FileCopyrightText: Copyright 2024, Arm Limited and/or its affiliates.
-# SPDX-License-Identifier: Apache-2.0
-"""Example rewrite with one fully connected clustered layer."""
-from typing import Any
-
-import tensorflow_model_optimization as tfmot
-from keras.api._v2 import keras  # Temporary workaround for now: MLIA-1107
-
-
-def get_keras_model_clus(input_shape: Any, output_shape: Any) -> keras.Model:
-    """Generate TensorFlow Lite model for clustering rewrite."""
-    rewrite_params = {
-        "number_of_clusters": 32,
-        "cluster_centroids_init": tfmot.clustering.keras.CentroidInitialization.LINEAR,
-    }
-    model = tfmot.clustering.keras.cluster_weights(
-        to_cluster=keras.Sequential(
-            [
-                keras.layers.InputLayer(input_shape=input_shape),
-                keras.layers.Flatten(),
-                keras.layers.Dense(units=output_shape),
-            ]
-        ),
-        **rewrite_params
-    )
-    return model