From 49dfd03617ed28a4fcbd847dd958c27442c84556 Mon Sep 17 00:00:00 2001 From: Annie Tallund Date: Mon, 6 Feb 2023 14:56:31 +0100 Subject: MLIA-709 Update compatibility data for Cortex-A - Add operator compatibility data for Cortex-A via for ArmNN TensorFlow Lite delegate 22.11 - Extend the Cortex-A target profile to include the version of the ArmNN TensorFlow Lite delegate to be used. - Some re-factoring work to support multiple versions and the new target profile parameter. Change-Id: Iae91bb0757ea3909be975af68b34d0ca2be47c43 --- src/mlia/target/cortex_a/advisor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'src/mlia/target/cortex_a/advisor.py') diff --git a/src/mlia/target/cortex_a/advisor.py b/src/mlia/target/cortex_a/advisor.py index a093784..5c077fb 100644 --- a/src/mlia/target/cortex_a/advisor.py +++ b/src/mlia/target/cortex_a/advisor.py @@ -36,11 +36,13 @@ class CortexAInferenceAdvisor(DefaultInferenceAdvisor): def get_collectors(self, context: Context) -> list[DataCollector]: """Return list of the data collectors.""" model = self.get_model(context) + target_profile = self.get_target_profile(context) + target_config = cast(CortexAConfiguration, profile(target_profile)) collectors: list[DataCollector] = [] if context.category_enabled(AdviceCategory.COMPATIBILITY): - collectors.append(CortexAOperatorCompatibility(model)) + collectors.append(CortexAOperatorCompatibility(model, target_config)) return collectors -- cgit v1.2.1