1.1 KiB
TensorFlow Lite operators that the Arm NN TensorFlow Lite Delegate supports
This reference guide provides a list of TensorFlow Lite operators the Arm NN SDK currently supports.
Fully supported
The Arm NN SDK TensorFlow Lite delegate currently supports the following operators:
-
ABS
-
ADD
-
AVERAGE_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
-
CONCATENATION, Supported Fused Activation: RELU , RELU6 , TANH, NONE
-
CONV_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
-
DEPTHWISE_CONV_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
-
DEQUANTIZE
-
DIV
-
EQUAL
-
EXP
-
FULLY_CONNECTED
-
GREATER
-
GREATER_OR_EQUAL
-
LESS
-
LESS_OR_EQUAL
-
LOGISTIC
-
LOG_SOFTMAX
-
L2_POOL_2D
-
MAXIMUM
-
MAX_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
-
MEAN
-
MINIMUM
-
MUL
-
NEG
-
NOT_EQUAL
-
QUANTIZE
-
RESHAPE
-
RESIZE_BILINEAR
-
RESIZE_NEAREST_NEIGHBOR
-
RELU
-
RELU6
-
RSQRT
-
SOFTMAX
-
SQRT
-
SUB
-
TANH
-
TRANSPOSE
-
TRANSPOSE_CONV
More machine learning operators will be supported in future releases.