1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
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
},
},
}
|
