Add a CLI component to enable rewrites

* Add flags for rewrite (--rewrite, --rewrite-start,
  --rewrite-end, --rewrite-target)
* Refactor CLI interfaces to accept tflite models with optimize for
  rewrite, keras models with optimize for clustering and pruning
* Refactor and move common.py and select.py out of the folder
  nn/tensorflow/optimizations
* Add file nn/rewrite/core/rewrite.py as placeholder
* Update/add unit tests
* Refactor OptimizeModel in ethos_u/data_collection.py
  for accepting tflite model case
* Extend the logic so that if "--rewrite" is specified, we don't add
  pruning to also accept TFLite models.
* Update README.md

Resolves: MLIA-750, MLIA-854, MLIA-865
Signed-off-by: Benjamin Klimczak <benjamin.klimczak@arm.com>
Change-Id: I67d85f71fa253d2bad4efe304ad8225970b9622c

1 files changed, 29 insertions, 0 deletions

diff --git a/src/mlia/nn/common.py b/src/mlia/nn/common.py
new file mode 100644
index 0000000..5a8d685
--- /dev/null
+++ b/src/mlia/nn/common.py
@@ -0,0 +1,29 @@
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Common items for the optimizations module."""
+from abc import ABC
+from abc import abstractmethod
+from dataclasses import dataclass
+
+from mlia.nn.tensorflow.config import ModelConfiguration
+
+
+@dataclass
+class OptimizerConfiguration:
+    """Abstract optimizer configuration."""
+
+
+class Optimizer(ABC):
+    """Abstract class for the optimizer."""
+
+    @abstractmethod
+    def get_model(self) -> ModelConfiguration:
+        """Abstract method to return the model instance from the optimizer."""
+
+    @abstractmethod
+    def apply_optimization(self) -> None:
+        """Abstract method to apply optimization to the model."""
+
+    @abstractmethod
+    def optimization_config(self) -> str:
+        """Return string representation of the optimization config."""