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, 45 insertions, 0 deletions

diff --git a/src/mlia/nn/rewrite/core/rewrite.py b/src/mlia/nn/rewrite/core/rewrite.py
new file mode 100644
index 0000000..d4f61c5
--- /dev/null
+++ b/src/mlia/nn/rewrite/core/rewrite.py
@@ -0,0 +1,45 @@
+# SPDX-FileCopyrightText: Copyright 2023, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Contains class Rewriter to replace a subgraph/layer of a model."""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+
+from mlia.nn.common import Optimizer
+from mlia.nn.common import OptimizerConfiguration
+from mlia.nn.tensorflow.config import TFLiteModel
+
+
+@dataclass
+class RewriteConfiguration(OptimizerConfiguration):
+    """Rewrite configuration."""
+
+    optimization_target: str
+    layers_to_optimize: list[str] | None = None
+    dataset: Path | None = None
+
+    def __str__(self) -> str:
+        """Return string representation of the configuration."""
+        return f"rewrite: {self.optimization_target}"
+
+
+class Rewriter(Optimizer):
+    """Rewriter class for basic rewrite flow."""
+
+    def __init__(
+        self, tflite_model_path: Path, optimizer_configuration: RewriteConfiguration
+    ):
+        """Init Rewriter instance."""
+        self.model = TFLiteModel(tflite_model_path)
+        self.optimizer_configuration = optimizer_configuration
+
+    def apply_optimization(self) -> None:
+        """Apply the rewrite flow."""
+
+    def get_model(self) -> TFLiteModel:
+        """Return optimized model."""
+        return self.model
+
+    def optimization_config(self) -> str:
+        """Optimization configirations."""