aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/InternalTypes.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/InternalTypes.hpp')
-rw-r--r--src/armnn/InternalTypes.hpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/InternalTypes.hpp b/src/armnn/InternalTypes.hpp
index e95a63af45..778408ae60 100644
--- a/src/armnn/InternalTypes.hpp
+++ b/src/armnn/InternalTypes.hpp
@@ -74,7 +74,8 @@
X(Subtraction) \
X(Switch) \
X(Transpose) \
- X(TransposeConvolution2d)
+ X(TransposeConvolution2d) \
+ X(Unmap)
/// When adding a new layer, adapt also the LastLayer enum value in the
/// enum class LayerType below
@@ -87,7 +88,7 @@ enum class LayerType
LIST_OF_LAYER_TYPE
#undef X
FirstLayer = Activation,
- LastLayer = TransposeConvolution2d
+ LastLayer = Unmap
};
const char* GetLayerTypeAsCString(LayerType type);