The layers that ArmNN SDK Serializer currently supports.
This reference guide provides a list of layers which can be serialized currently by the Arm NN SDK.
Fully supported
The Arm NN SDK Serializer currently supports the following layers:
- Activation
- Addition
- BatchToSpaceNd
- BatchNormalization
- Constant
- Convolution2d
- DepthwiseConvolution2d
- Dequantize
- DetectionPostProcess
- Division
- Equal
- Floor
- FullyConnected
- Gather
- Greater
- L2Normalization
- Lstm
- Maximum
- Mean
- Merge
- Merger
- Minimum
- Multiplication
- Normalization
- Pad
- Permute
- Pooling2d
- Reshape
- ResizeBilinear
- Rsqrt
- Softmax
- SpaceToBatchNd
- Splitter
- StridedSlice
- Subtraction
- Switch
More machine learning layers will be supported in future releases.