aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/ethosu/reporters.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-10-24 15:08:08 +0100
committerDmitrii Agibov <dmitrii.agibov@arm.com>2022-10-26 17:08:13 +0100
commit58a65fee574c00329cf92b387a6d2513dcbf6100 (patch)
tree47e3185f78b4298ab029785ddee68456e44cac10 /src/mlia/devices/ethosu/reporters.py
parent9d34cb72d45a6d0a2ec1063ebf32536c1efdba75 (diff)
downloadmlia-58a65fee574c00329cf92b387a6d2513dcbf6100.tar.gz
MLIA-433 Add TensorFlow Lite compatibility check
- Add ability to intercept low level TensorFlow output - Produce advice for the models that could not be converted to the TensorFlow Lite format - Refactor utility functions for TensorFlow Lite conversion - Add TensorFlow Lite compatibility checker Change-Id: I47d120d2619ced7b143bc92c5184515b81c0220d
Diffstat (limited to 'src/mlia/devices/ethosu/reporters.py')
-rw-r--r--src/mlia/devices/ethosu/reporters.py14
1 files changed, 1 insertions, 13 deletions
diff --git a/src/mlia/devices/ethosu/reporters.py b/src/mlia/devices/ethosu/reporters.py
index 9181043..f0fcb39 100644
--- a/src/mlia/devices/ethosu/reporters.py
+++ b/src/mlia/devices/ethosu/reporters.py
@@ -8,6 +8,7 @@ from typing import Any
from typing import Callable
from mlia.core.advice_generation import Advice
+from mlia.core.reporters import report_advice
from mlia.core.reporting import BytesCell
from mlia.core.reporting import Cell
from mlia.core.reporting import ClockCell
@@ -360,19 +361,6 @@ def report_perf_metrics(
)
-def report_advice(advice: list[Advice]) -> Report:
- """Generate report for the advice."""
- return Table(
- columns=[
- Column("#", only_for=["plain_text"]),
- Column("Advice", alias="advice_message"),
- ],
- rows=[(i + 1, a.messages) for i, a in enumerate(advice)],
- name="Advice",
- alias="advice",
- )
-
-
def ethos_u_formatters(data: Any) -> Callable[[Any], Report]:
"""Find appropriate formatter for the provided data."""
if isinstance(data, PerformanceMetrics) or is_list_of(data, PerformanceMetrics, 2):