• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# TensorFlow Lite operators that the Arm NN SDK supports
2
3This reference guide provides a list of TensorFlow Lite operators the Arm NN SDK currently supports.
4
5## Fully supported
6
7The Arm NN SDK TensorFlow Lite parser currently supports the following operators:
8
9* ADD
10
11* AVERAGE_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
12
13* BATCH_TO_SPACE
14
15* CONCATENATION, Supported Fused Activation: RELU , RELU6 , TANH, NONE
16
17* CONV_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
18
19* DEPTHWISE_CONV_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
20
21* DEQUANTIZE
22
23* DIV
24
25* EXP
26
27* FULLY_CONNECTED, Supported Fused Activation: RELU , RELU6 , TANH, NONE
28
29* LEAKY_RELU
30
31* LOGISTIC
32
33* L2_NORMALIZATION
34
35* MAX_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE
36
37* MAXIMUM
38
39* MEAN
40
41* MINIMUM
42
43* MUL
44
45* NEG
46
47* PACK
48
49* PAD
50
51* QUANTIZE
52
53* RELU
54
55* RELU6
56
57* RESHAPE
58
59* RESIZE_BILINEAR
60
61* RESIZE_NEAREST_NEIGHBOR
62
63* SLICE
64
65* SOFTMAX
66
67* SPACE_TO_BATCH
68
69* SPLIT
70
71* SPLIT_V
72
73* SQUEEZE
74
75* STRIDED_SLICE
76
77* SUB
78
79* TANH
80
81* TRANSPOSE
82
83* TRANSPOSE_CONV
84
85* UNPACK
86
87## Custom Operator
88
89* TFLite_Detection_PostProcess
90
91## Tested networks
92
93Arm tested these operators with the following TensorFlow Lite neural network:
94
95* [Quantized MobileNet](http://download.tensorflow.org/models/mobilenet_v1_2018_02_22/mobilenet_v1_1.0_224_quant.tgz)
96
97* [Quantized SSD MobileNet](http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18.tar.gz)
98
99* DeepSpeech v1 converted from [TensorFlow model](https://github.com/mozilla/DeepSpeech/releases/tag/v0.4.1)
100
101* DeepSpeaker
102
103* [DeepLab v3+](https://www.tensorflow.org/lite/models/segmentation/overview)
104
105* FSRCNN
106
107* EfficientNet-lite
108
109* RDN converted from [TensorFlow model](https://github.com/hengchuan/RDN-TensorFlow)
110
111* Quantized RDN (CpuRef)
112
113* [Quantized Inception v3](http://download.tensorflow.org/models/tflite_11_05_08/inception_v3_quant.tgz)
114
115* [Quantized Inception v4](http://download.tensorflow.org/models/inception_v4_299_quant_20181026.tgz) (CpuRef)
116
117* Quantized ResNet v2 50 (CpuRef)
118
119* Quantized Yolo v3 (CpuRef)
120
121More machine learning operators will be supported in future releases.
122