diff options
Diffstat (limited to 'src/mlia/target/cortex_a/config.py')
-rw-r--r-- | src/mlia/target/cortex_a/config.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/src/mlia/target/cortex_a/config.py b/src/mlia/target/cortex_a/config.py index fd39e0a..f91031e 100644 --- a/src/mlia/target/cortex_a/config.py +++ b/src/mlia/target/cortex_a/config.py @@ -5,6 +5,7 @@ from __future__ import annotations from typing import Any +from mlia.backend.armnn_tflite_delegate.compat import ARMNN_TFLITE_DELEGATE from mlia.target.config import TargetProfile @@ -16,8 +17,22 @@ class CortexAConfiguration(TargetProfile): target = kwargs["target"] super().__init__(target) + self.backend_config = kwargs.get("backend") + self.armnn_tflite_delegate_version = kwargs["backend"]["armnn-tflite-delegate"][ + "version" + ] + def verify(self) -> None: """Check the parameters.""" super().verify() if self.target != "cortex-a": raise ValueError(f"Wrong target {self.target} for Cortex-A configuration.") + + if not self.armnn_tflite_delegate_version: + raise ValueError("No version for ArmNN TensorFlow Lite delegate specified.") + if self.armnn_tflite_delegate_version not in ARMNN_TFLITE_DELEGATE["ops"]: + raise ValueError( + f"Version '{self.armnn_tflite_delegate_version}' of " + f"backend {ARMNN_TFLITE_DELEGATE['backend']} is not supported. " + f"Choose from: {', '.join(ARMNN_TFLITE_DELEGATE['ops'])}" + ) |