This reference guide provides a list of TensorFlow Lite operators the Arm NN SDK currently supports.
The Arm NN SDK TensorFlow Lite parser currently only supports uint8.
The Arm NN SDK TensorFlow Lite parser currently supports the following operators:
ADD
AVERAGE_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
BATCH_TO_SPACE
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
FULLY_CONNECTED, Supported Fused Activation: RELU , RELU6 , TANH, NONE
LOGISTIC
MAX_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
MAXIMUM
MEAN
MINIMUM
MUL
PAD
RELU
RELU6
RESHAPE
RESIZE_BILINEAR
SOFTMAX
SPACE_TO_BATCH
SPLIT
SQUEEZE
STRIDED_SLICE
SUB
Arm tested these operators with the following TensorFlow Lite neural network:
More machine learning operators will be supported in future releases.