diff options
Diffstat (limited to 'src/mlia/backend/armnn_tflite_delegate/ops/v23_05.py')
-rw-r--r-- | src/mlia/backend/armnn_tflite_delegate/ops/v23_05.py | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/src/mlia/backend/armnn_tflite_delegate/ops/v23_05.py b/src/mlia/backend/armnn_tflite_delegate/ops/v23_05.py new file mode 100644 index 0000000..ff41820 --- /dev/null +++ b/src/mlia/backend/armnn_tflite_delegate/ops/v23_05.py @@ -0,0 +1,125 @@ +# SPDX-FileCopyrightText: Copyright 2023, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +"""Collection of Cortex-A operator compatibility information.""" +from __future__ import annotations + +# pylint: disable=duplicate-code + +VERSION = "23.05" +SUPPORTED_FUSED_ACTIVATION_FUNCTIONS = [ + "NONE", + "RELU", + "RELU6", + "RELU_N1_TO_1", + "SIGMOID", + "TANH", +] +OPERATORS = { + "builtin_ops": { + "ABS": {}, + "ADD": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "ARG_MAX": {}, + "ARG_MIN": {}, + "AVERAGE_POOL_2D": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "BATCH_MATMUL": {}, + "BATCH_TO_SPACE_ND": {}, + "CAST": {}, + "CEIL": {}, + "CONCATENATION": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "CONV_2D": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "CONV_3D": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "DEPTH_TO_SPACE": {}, + "DEPTHWISE_CONV_2D": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "DEQUANTIZE": {}, + "DIV": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "ELU": {}, + "EQUAL": {}, + "EXP": {}, + "EXPAND_DIMS": {}, + "FILL": {}, + "FLOOR": {}, + "FLOOR_DIV": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "FULLY_CONNECTED": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "GATHER": {}, + "GATHER_ND": {}, + "GREATER": {}, + "GREATER_EQUAL": {}, + "HARD_SWISH": {}, + "L2_NORMALIZATION": {}, + "L2_POOL_2D": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "LESS": {}, + "LESS_EQUAL": {}, + "LOCAL_RESPONSE_NORMALIZATION": {}, + "LOG": {}, + "LOGICAL_AND": {}, + "LOGICAL_NOT": {}, + "LOGICAL_OR": {}, + "LOGISTIC": {}, + "LOG_SOFTMAX": {}, + "LSTM": {}, + "MAX_POOL_2D": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "MAXIMUM": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "MEAN": {}, + "MINIMUM": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "MIRROR_PAD": {}, + "MUL": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "NEG": {}, + "NOT_EQUAL": {}, + "PACK": {}, + "PAD": {}, + "PADV2": {}, + "PRELU": {}, + "QUANTIZE": {}, + "REDUCE_MAX": {}, + "REDUCE_MIN": {}, + "REDUCE_PROD": {}, + "RELU": {}, + "RELU_N1_TO_1": {}, + "RELU6": {}, + "RESHAPE": {}, + "RESIZE_BILINEAR": {}, + "RESIZE_NEAREST_NEIGHBOR": {}, + "RSQRT": {}, + "SHAPE": {}, + "SIN": {}, + "SLICE": {}, + "SOFTMAX": {}, + "SPACE_TO_BATCH_ND": {}, + "SPACE_TO_DEPTH": {}, + "SPLIT": {}, + "SPLIT_V": {}, + "SQRT": {}, + "SQUEEZE": {}, + "STRIDED_SLICE": {}, + "SUB": {"supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS}, + "SUM": {}, + "TANH": {}, + "TRANSPOSE": {}, + "TRANSPOSE_CONV": {}, + "UNIDIRECTIONAL_SEQUENCE_LSTM": {}, + "UNPACK": {}, + }, + # CUSTOM OPERATORS + "custom_ops": { + "AveragePool3D": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + "MaxPool3D": { + "supported_fused_activation": SUPPORTED_FUSED_ACTIVATION_FUNCTIONS + }, + }, +} |