MLCE-530 Add front end support for UnidirectionalSequenceLstm on ArmNN
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I57bcbdec3eb0155f41af0fe7d6abf9bac2ec86eb
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp
index e7c1760..056aa83 100644
--- a/include/armnn/Types.hpp
+++ b/include/armnn/Types.hpp
@@ -333,7 +333,6 @@
X(ArgMinMax) \
X(BatchNormalization) \
X(BatchToSpaceNd) \
- X(Cast) \
X(Comparison) \
X(Concat) \
X(Constant) \
@@ -382,7 +381,6 @@
X(Rank) \
X(Resize) \
X(Reduce) \
- X(Shape) \
X(Slice) \
X(Softmax) \
X(SpaceToBatchNd) \
@@ -396,6 +394,11 @@
X(Transpose) \
X(TransposeConvolution2d) \
X(Unmap) \
+ X(Cast) \
+ X(Shape) \
+ X(UnidirectionalSequenceLstm) \
+
+// New layers should be added at last to minimize instability.
/// When adding a new layer, adapt also the LastLayer enum value in the
/// enum class LayerType below
@@ -405,7 +408,7 @@
LIST_OF_LAYER_TYPE
#undef X
FirstLayer = Activation,
- LastLayer = Unmap
+ LastLayer = UnidirectionalSequenceLstm
};
const char* GetLayerTypeAsCString(LayerType type);