5 files changed, 54 insertions, 3 deletions
diff --git a/src/mlia/target/tosa/advice_generation.py b/src/mlia/target/tosa/advice_generation.py
index b8b9abf..ad321b2 100644
--- a/src/mlia/target/tosa/advice_generation.py
+++ b/src/mlia/target/tosa/advice_generation.py
@@ -7,6 +7,10 @@ from mlia.core.advice_generation import advice_category
 from mlia.core.advice_generation import FactBasedAdviceProducer
 from mlia.core.common import AdviceCategory
 from mlia.core.common import DataItem
+from mlia.target.common.reporters import handle_model_is_not_tflite_compatible_common
+from mlia.target.common.reporters import handle_tflite_check_failed_common
+from mlia.target.common.reporters import ModelIsNotTFLiteCompatible
+from mlia.target.common.reporters import TFLiteCompatibilityCheckFailed
 from mlia.target.tosa.data_analysis import ModelIsNotTOSACompatible
 from mlia.target.tosa.data_analysis import ModelIsTOSACompatible
 
@@ -38,3 +42,19 @@ class TOSAAdviceProducer(FactBasedAdviceProducer):
                 "Please, refer to the operators table for more information."
             ]
         )
+
+    @produce_advice.register
+    @advice_category(AdviceCategory.COMPATIBILITY)
+    def handle_model_is_not_tflite_compatible(
+        self, data_item: ModelIsNotTFLiteCompatible
+    ) -> None:
+        """Advice for TensorFlow Lite compatibility."""
+        handle_model_is_not_tflite_compatible_common(self, data_item)
+
+    @produce_advice.register
+    @advice_category(AdviceCategory.COMPATIBILITY)
+    def handle_tflite_check_failed(
+        self, _data_item: TFLiteCompatibilityCheckFailed
+    ) -> None:
+        """Advice for the failed TensorFlow Lite compatibility checks."""
+        handle_tflite_check_failed_common(self, _data_item)
diff --git a/src/mlia/target/tosa/data_analysis.py b/src/mlia/target/tosa/data_analysis.py
index 7cbd61d..7b31441 100644
--- a/src/mlia/target/tosa/data_analysis.py
+++ b/src/mlia/target/tosa/data_analysis.py
@@ -1,6 +1,8 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """TOSA data analysis module."""
+from __future__ import annotations
+
 from dataclasses import dataclass
 from functools import singledispatchmethod
 
@@ -8,6 +10,8 @@ from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo
 from mlia.core.common import DataItem
 from mlia.core.data_analysis import Fact
 from mlia.core.data_analysis import FactExtractor
+from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo
+from mlia.target.common.reporters import analyze_tflite_compatibility_common
 
 
 @dataclass
@@ -34,3 +38,8 @@ class TOSADataAnalyzer(FactExtractor):
             self.add_fact(ModelIsTOSACompatible())
         else:
             self.add_fact(ModelIsNotTOSACompatible())
+
+    @analyze_data.register
+    def analyze_tflite_compatibility(self, data_item: TFLiteCompatibilityInfo) -> None:
+        """Analyze TensorFlow Lite compatibility information."""
+        analyze_tflite_compatibility_common(self, data_item)
diff --git a/src/mlia/target/tosa/data_collection.py b/src/mlia/target/tosa/data_collection.py
index 105c501..19ea6eb 100644
--- a/src/mlia/target/tosa/data_collection.py
+++ b/src/mlia/target/tosa/data_collection.py
@@ -1,12 +1,17 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """TOSA data collection module."""
+from __future__ import annotations
+
 from pathlib import Path
 
 from mlia.backend.tosa_checker.compat import get_tosa_compatibility_info
 from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo
 from mlia.core.data_collection import ContextAwareDataCollector
 from mlia.nn.tensorflow.config import get_tflite_model
+from mlia.nn.tensorflow.tflite_compat import TFLiteChecker
+from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo
+from mlia.nn.tensorflow.utils import is_tflite_model
 from mlia.utils.logging import log_action
 
 
@@ -17,8 +22,16 @@ class TOSAOperatorCompatibility(ContextAwareDataCollector):
         """Init the data collector."""
         self.model = model
 
-    def collect_data(self) -> TOSACompatibilityInfo:
+    def collect_data(self) -> TFLiteCompatibilityInfo | TOSACompatibilityInfo | None:
         """Collect TOSA compatibility information."""
+        if not is_tflite_model(self.model):
+            with log_action("Checking TensorFlow Lite compatibility ..."):
+                tflite_checker = TFLiteChecker()
+                tflite_compat = tflite_checker.check_compatibility(self.model)
+
+            if not tflite_compat.compatible:
+                return tflite_compat
+
         tflite_model = get_tflite_model(self.model, self.context)
 
         with log_action("Checking operator compatibility ..."):
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)
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}")