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
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
MEAN
MUL
PAD
RELU
RELU6
RESHAPE
SOFTMAX
SQUEEZE
Arm tested these operators with the following TensorFlow Lite neural network:
More machine learning operators will be supported in future releases.