diff options
author | Annie Tallund <annie.tallund@arm.com> | 2023-02-06 14:56:31 +0100 |
---|---|---|
committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-02-13 16:20:41 +0000 |
commit | 49dfd03617ed28a4fcbd847dd958c27442c84556 (patch) | |
tree | 1fe14ba3980dd6be15f866e97b67582831344eb9 /tests/test_target_cortex_a_advice_generation.py | |
parent | e3bef50932abdc2e32fa5636fb7a496b149e72d9 (diff) | |
download | mlia-49dfd03617ed28a4fcbd847dd958c27442c84556.tar.gz |
MLIA-709 Update compatibility data for Cortex-A0.6.0-rc.1
- 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
Diffstat (limited to 'tests/test_target_cortex_a_advice_generation.py')
-rw-r--r-- | tests/test_target_cortex_a_advice_generation.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/test_target_cortex_a_advice_generation.py b/tests/test_target_cortex_a_advice_generation.py index b9edbb5..9596d47 100644 --- a/tests/test_target_cortex_a_advice_generation.py +++ b/tests/test_target_cortex_a_advice_generation.py @@ -14,15 +14,16 @@ from mlia.core.common import DataItem from mlia.core.context import ExecutionContext from mlia.nn.tensorflow.tflite_graph import TFL_ACTIVATION_FUNCTION from mlia.target.cortex_a.advice_generation import CortexAAdviceProducer +from mlia.target.cortex_a.config import CortexAConfiguration from mlia.target.cortex_a.data_analysis import ModelHasCustomOperators from mlia.target.cortex_a.data_analysis import ModelIsCortexACompatible from mlia.target.cortex_a.data_analysis import ModelIsNotCortexACompatible from mlia.target.cortex_a.data_analysis import ModelIsNotTFLiteCompatible from mlia.target.cortex_a.data_analysis import TFLiteCompatibilityCheckFailed +VERSION = CortexAConfiguration.load_profile("cortex-a").armnn_tflite_delegate_version BACKEND_INFO = ( - f"{ARMNN_TFLITE_DELEGATE['metadata']['backend']} " - f"{ARMNN_TFLITE_DELEGATE['metadata']['version']}" + f"{ARMNN_TFLITE_DELEGATE['backend']} " f"{ARMNN_TFLITE_DELEGATE['ops'][VERSION]}" ) |