MLIA-711 Extend TensorFlow Lite Compatibility Check

- Unify the TensorFlow Lite compatibility check across Cortex-A, TOSA
and Ethos-U targets
- Display tables/messages with parsed information
- Do not display raw TensorFlow Lite errors, and return with exit code 0

Change-Id: I9333fdb6cbe592f1ed7395d392412168492a1479

1 files changed, 5 insertions, 0 deletions

diff --git a/src/mlia/target/tosa/reporters.py b/src/mlia/target/tosa/reporters.py
index f54c06b..decae0c 100644
--- a/src/mlia/target/tosa/reporters.py
+++ b/src/mlia/target/tosa/reporters.py
@@ -22,6 +22,8 @@ from mlia.core.reporting import NestedReport
 from mlia.core.reporting import Report
 from mlia.core.reporting import ReportItem
 from mlia.core.reporting import Table
+from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo
+from mlia.target.common.reporters import report_tflite_compatiblity
 from mlia.target.tosa.config import TOSAConfiguration
 from mlia.target.tosa.metadata import TOSAMetadata
 from mlia.utils.console import style_improvement
@@ -163,4 +165,7 @@ def tosa_formatters(data: Any) -> Callable[[Any], Report]:
     if isinstance(data, TOSACompatibilityInfo):
         return report_tosa_compatibility
 
+    if isinstance(data, TFLiteCompatibilityInfo):
+        return report_tflite_compatiblity  # type: ignore
+
     raise Exception(f"Unable to find appropriate formatter for {data}")