aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/cortexa/handlers.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/devices/cortexa/handlers.py')
-rw-r--r--src/mlia/devices/cortexa/handlers.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/mlia/devices/cortexa/handlers.py b/src/mlia/devices/cortexa/handlers.py
index f54ceff..7ed2b75 100644
--- a/src/mlia/devices/cortexa/handlers.py
+++ b/src/mlia/devices/cortexa/handlers.py
@@ -12,6 +12,7 @@ from mlia.devices.cortexa.events import CortexAAdvisorEventHandler
from mlia.devices.cortexa.events import CortexAAdvisorStartedEvent
from mlia.devices.cortexa.operators import CortexACompatibilityInfo
from mlia.devices.cortexa.reporters import cortex_a_formatters
+from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo
logger = logging.getLogger(__name__)
@@ -30,6 +31,9 @@ class CortexAEventHandler(WorkflowEventsHandler, CortexAAdvisorEventHandler):
if isinstance(data_item, CortexACompatibilityInfo):
self.reporter.submit(data_item.operators, delay_print=True)
+ if isinstance(data_item, TFLiteCompatibilityInfo) and not data_item.compatible:
+ self.reporter.submit(data_item, delay_print=True)
+
def on_cortex_a_advisor_started(self, event: CortexAAdvisorStartedEvent) -> None:
"""Handle CortexAAdvisorStarted event."""
self.reporter.submit(event.device)