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

2 files changed, 350 insertions, 177 deletions

diff --git a/src/mlia/backend/armnn_tflite_delegate/__init__.py b/src/mlia/backend/armnn_tflite_delegate/__init__.py
index 4fe8639..66e5a2a 100644
--- a/src/mlia/backend/armnn_tflite_delegate/__init__.py
+++ b/src/mlia/backend/armnn_tflite_delegate/__init__.py
@@ -1,6 +1,8 @@
 # SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """Arm NN TensorFlow Lite delegate backend module."""
+from typing import cast
+
 from mlia.backend.armnn_tflite_delegate.compat import ARMNN_TFLITE_DELEGATE
 from mlia.backend.config import BackendConfiguration
 from mlia.backend.config import BackendType
@@ -14,5 +16,5 @@ registry.register(
         supported_systems=None,
         backend_type=BackendType.BUILTIN,
     ),
-    pretty_name=ARMNN_TFLITE_DELEGATE["metadata"]["backend"],
+    pretty_name=cast(str, ARMNN_TFLITE_DELEGATE["backend"]),
 )
diff --git a/src/mlia/backend/armnn_tflite_delegate/compat.py b/src/mlia/backend/armnn_tflite_delegate/compat.py
index c474e75..e2650be 100644
--- a/src/mlia/backend/armnn_tflite_delegate/compat.py
+++ b/src/mlia/backend/armnn_tflite_delegate/compat.py
@@ -1,184 +1,355 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """Collection of Cortex-A operator compatibility information."""
 from __future__ import annotations
 
-from typing import Any
-
-ARMNN_TFLITE_DELEGATE: dict[str, dict[str, Any]] = {
-    "metadata": {
-        "backend": "Arm NN TensorFlow Lite delegate",
-        "version": "22.08",
-    },
-    # BUILTIN OPERATORS
-    "builtin_ops": {
-        "ABS": {},
-        "ADD": {},
-        "ARG_MAX": {},
-        "ARG_MIN": {},
-        "AVERAGE_POOL_2D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "BATCH_TO_SPACE_ND": {},
-        "CAST": {},
-        "CONCATENATION": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "CONV_2D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "CONV_3D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "DEPTH_TO_SPACE": {},
-        "DEPTHWISE_CONV_2D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "DEQUANTIZE": {},
-        "DIV": {},
-        "EQUAL": {},
-        "ELU": {},
-        "EXP": {},
-        "EXPAND_DIMS": {},
-        "FILL": {},
-        "FLOOR": {},
-        "FLOOR_DIV": {},
-        "FULLY_CONNECTED": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "GATHER": {},
-        "GATHER_ND": {},
-        "GREATER": {},
-        "GREATER_EQUAL": {},
-        "HARD_SWISH": {},
-        "L2_NORMALIZATION": {},
-        "L2_POOL_2D": {},
-        "LESS": {},
-        "LESS_EQUAL": {},
-        "LOCAL_RESPONSE_NORMALIZATION": {},
-        "LOG": {},
-        "LOGICAL_AND": {},
-        "LOGICAL_NOT": {},
-        "LOGICAL_OR": {},
-        "LOGISTIC": {},
-        "LOG_SOFTMAX": {},
-        "LSTM": {},
-        "MAXIMUM": {},
-        "MAX_POOL_2D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "TANH",
-                "NONE",
-            ]
-        },
-        "MEAN": {},
-        "MINIMUM": {},
-        "MIRROR_PAD": {},
-        "MUL": {},
-        "NEG": {},
-        "NOT_EQUAL": {},
-        "PACK": {},
-        "PAD": {},
-        "PADV2": {},
-        "PRELU": {},
-        "QUANTIZE": {},
-        "RANK": {},
-        "REDUCE_MAX": {},
-        "REDUCE_MIN": {},
-        "REDUCE_PROD": {},
-        "RELU": {},
-        "RELU6": {},
-        "RELU_N1_TO_1": {},
-        "RESHAPE": {},
-        "RESIZE_BILINEAR": {},
-        "RESIZE_NEAREST_NEIGHBOR": {},
-        "RSQRT": {},
-        "SHAPE": {},
-        "SIN": {},
-        "SOFTMAX": {},
-        "SPACE_TO_BATCH_ND": {},
-        "SPACE_TO_DEPTH": {},
-        "SPLIT": {},
-        "SPLIT_V": {},
-        "SQRT": {},
-        "SQUEEZE": {},
-        "STRIDED_SLICE": {},
-        "SUB": {},
-        "SUM": {},
-        "TANH": {},
-        "TRANSPOSE": {},
-        "TRANSPOSE_CONV": {},
-        "UNIDIRECTIONAL_SEQUENCE_LSTM": {},
-        "UNPACK": {},
-    },
-    # CUSTOM OPERATORS
-    "custom_ops": {
-        "AveragePool3D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "SIGN_BIT",
-                "TANH",
-                "NONE",
-            ]
+ARMNN_TFLITE_DELEGATE: dict = {
+    "backend": "Arm NN TensorFlow Lite delegate",
+    "ops": {
+        "22.08": {
+            "builtin_ops": {
+                "ABS": {},
+                "ADD": {},
+                "ARG_MAX": {},
+                "ARG_MIN": {},
+                "AVERAGE_POOL_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "BATCH_TO_SPACE_ND": {},
+                "CAST": {},
+                "CONCATENATION": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "CONV_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "CONV_3D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "DEPTH_TO_SPACE": {},
+                "DEPTHWISE_CONV_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "DEQUANTIZE": {},
+                "DIV": {},
+                "ELU": {},
+                "EQUAL": {},
+                "EXP": {},
+                "EXPAND_DIMS": {},
+                "FILL": {},
+                "FLOOR": {},
+                "FLOOR_DIV": {},
+                "FULLY_CONNECTED": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "GATHER": {},
+                "GATHER_ND": {},
+                "GREATER": {},
+                "GREATER_EQUAL": {},
+                "HARD_SWISH": {},
+                "L2_NORMALIZATION": {},
+                "L2_POOL_2D": {},
+                "LESS": {},
+                "LESS_EQUAL": {},
+                "LOCAL_RESPONSE_NORMALIZATION": {},
+                "LOG": {},
+                "LOGICAL_AND": {},
+                "LOGICAL_NOT": {},
+                "LOGICAL_OR": {},
+                "LOGISTIC": {},
+                "LOG_SOFTMAX": {},
+                "LSTM": {},
+                "MAX_POOL_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "MAXIMUM": {},
+                "MEAN": {},
+                "MINIMUM": {},
+                "MIRROR_PAD": {},
+                "MUL": {},
+                "NEG": {},
+                "NOT_EQUAL": {},
+                "PACK": {},
+                "PAD": {},
+                "PADV2": {},
+                "PRELU": {},
+                "QUANTIZE": {},
+                "RANK": {},
+                "REDUCE_MAX": {},
+                "REDUCE_MIN": {},
+                "REDUCE_PROD": {},
+                "RELU": {},
+                "RELU_N1_TO_1": {},
+                "RELU6": {},
+                "RESHAPE": {},
+                "RESIZE_BILINEAR": {},
+                "RESIZE_NEAREST_NEIGHBOR": {},
+                "RSQRT": {},
+                "SHAPE": {},
+                "SIN": {},
+                "SOFTMAX": {},
+                "SPACE_TO_BATCH_ND": {},
+                "SPACE_TO_DEPTH": {},
+                "SPLIT": {},
+                "SPLIT_V": {},
+                "SQRT": {},
+                "SQUEEZE": {},
+                "STRIDED_SLICE": {},
+                "SUB": {},
+                "SUM": {},
+                "TANH": {},
+                "TRANSPOSE": {},
+                "TRANSPOSE_CONV": {},
+                "UNIDIRECTIONAL_SEQUENCE_LSTM": {},
+                "UNPACK": {},
+            },
+            # CUSTOM OPERATORS
+            "custom_ops": {
+                "AveragePool3D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "SIGN_BIT",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "MaxPool3D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "SIGN_BIT",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+            },
         },
-        "MaxPool3D": {
-            "supported_fused_activation": [
-                "RELU",
-                "RELU6",
-                "RELU_N1_TO_1",
-                "SIGMOID",
-                "SIGN_BIT",
-                "TANH",
-                "NONE",
-            ]
+        "22.11": {
+            "builtin_ops": {
+                "ABS": {},
+                "ADD": {},
+                "ARG_MAX": {},
+                "ARG_MIN": {},
+                "AVERAGE_POOL_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "BATCH_MATMUL": {},
+                "BATCH_TO_SPACE_ND": {},
+                "CAST": {},
+                "CONCATENATION": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "CONV_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "CONV_3D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "DEPTH_TO_SPACE": {},
+                "DEPTHWISE_CONV_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "DEQUANTIZE": {},
+                "DIV": {},
+                "ELU": {},
+                "EQUAL": {},
+                "EXP": {},
+                "EXPAND_DIMS": {},
+                "FILL": {},
+                "FLOOR": {},
+                "FLOOR_DIV": {},
+                "FULLY_CONNECTED": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "GATHER": {},
+                "GATHER_ND": {},
+                "GREATER": {},
+                "GREATER_EQUAL": {},
+                "HARD_SWISH": {},
+                "L2_NORMALIZATION": {},
+                "L2_POOL_2D": {},
+                "LESS": {},
+                "LESS_EQUAL": {},
+                "LOCAL_RESPONSE_NORMALIZATION": {},
+                "LOG": {},
+                "LOGICAL_AND": {},
+                "LOGICAL_NOT": {},
+                "LOGICAL_OR": {},
+                "LOGISTIC": {},
+                "LOG_SOFTMAX": {},
+                "LSTM": {},
+                "MAX_POOL_2D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "MAXIMUM": {},
+                "MEAN": {},
+                "MINIMUM": {},
+                "MIRROR_PAD": {},
+                "MUL": {},
+                "NEG": {},
+                "NOT_EQUAL": {},
+                "PACK": {},
+                "PAD": {},
+                "PADV2": {},
+                "PRELU": {},
+                "QUANTIZE": {},
+                "RANK": {},
+                "REDUCE_MAX": {},
+                "REDUCE_MIN": {},
+                "REDUCE_PROD": {},
+                "RELU": {},
+                "RELU_N1_TO_1": {},
+                "RELU6": {},
+                "RESHAPE": {},
+                "RESIZE_BILINEAR": {},
+                "RESIZE_NEAREST_NEIGHBOR": {},
+                "RSQRT": {},
+                "SHAPE": {},
+                "SIN": {},
+                "SOFTMAX": {},
+                "SPACE_TO_BATCH_ND": {},
+                "SPACE_TO_DEPTH": {},
+                "SPLIT": {},
+                "SPLIT_V": {},
+                "SQRT": {},
+                "SQUEEZE": {},
+                "STRIDED_SLICE": {},
+                "SUB": {},
+                "SUM": {},
+                "TANH": {},
+                "TRANSPOSE": {},
+                "TRANSPOSE_CONV": {},
+                "UNIDIRECTIONAL_SEQUENCE_LSTM": {},
+                "UNPACK": {},
+            },
+            # CUSTOM OPERATORS
+            "custom_ops": {
+                "AveragePool3D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "SIGN_BIT",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+                "MaxPool3D": {
+                    "supported_fused_activation": [
+                        "RELU",
+                        "RELU6",
+                        "RELU_N1_TO_1",
+                        "SIGMOID",
+                        "SIGN_BIT",
+                        "TANH",
+                        "NONE",
+                    ]
+                },
+            },
         },
     },
 }