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

diff --git a/src/mlia/target/tosa/handlers.py b/src/mlia/target/tosa/handlers.py
index 131afa7..26e7226 100644
--- a/src/mlia/target/tosa/handlers.py
+++ b/src/mlia/target/tosa/handlers.py
@@ -9,6 +9,7 @@ import logging
 from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo
 from mlia.core.events import CollectedDataEvent
 from mlia.core.handlers import WorkflowEventsHandler
+from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo
 from mlia.target.tosa.events import TOSAAdvisorEventHandler
 from mlia.target.tosa.events import TOSAAdvisorStartedEvent
 from mlia.target.tosa.reporters import tosa_formatters
@@ -34,3 +35,6 @@ class TOSAEventHandler(WorkflowEventsHandler, TOSAAdvisorEventHandler):
 
         if isinstance(data_item, TOSACompatibilityInfo):
             self.reporter.submit(data_item, delay_print=True)
+
+        if isinstance(data_item, TFLiteCompatibilityInfo) and not data_item.compatible:
+            self.reporter.submit(data_item, delay_print=True)