From e69fcc9ec42788218ba9693e53834b4606cf3882 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Thu, 12 Dec 2019 16:41:43 +0000 Subject: IVGCVSW-4267 Add missing layers to GetLayerTypeAsCString Change-Id: I8c017a5222ca58a6fae6e2fae9c483d29a5cc76d Signed-off-by: Jim Flynn --- src/armnn/InternalTypes.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/armnn/InternalTypes.cpp b/src/armnn/InternalTypes.cpp index 9b46436ffc..fea6cb3346 100644 --- a/src/armnn/InternalTypes.cpp +++ b/src/armnn/InternalTypes.cpp @@ -40,6 +40,7 @@ char const* GetLayerTypeAsCString(LayerType type) case LayerType::Maximum: return "Maximum"; case LayerType::Mean: return "Mean"; case LayerType::MemCopy: return "MemCopy"; + case LayerType::MemImport: return "MemImport"; case LayerType::Merge: return "Merge"; case LayerType::Minimum: return "Minimum"; case LayerType::Multiplication: return "Multiplication"; @@ -50,12 +51,14 @@ char const* GetLayerTypeAsCString(LayerType type) case LayerType::Pooling2d: return "Pooling2d"; case LayerType::PreCompiled: return "PreCompiled"; case LayerType::Prelu: return "Prelu"; + case LayerType::Quantize: return "Quantize"; case LayerType::QuantizedLstm: return "QuantizedLstm"; case LayerType::Reshape: return "Reshape"; - case LayerType::Rsqrt: return "Rsqrt"; case LayerType::Resize: return "Resize"; + case LayerType::Rsqrt: return "Rsqrt"; case LayerType::Softmax: return "Softmax"; case LayerType::SpaceToBatchNd: return "SpaceToBatchNd"; + case LayerType::SpaceToDepth: return "SpaceToDepth"; case LayerType::Splitter: return "Splitter"; case LayerType::Stack: return "Stack"; case LayerType::StridedSlice: return "StridedSlice"; -- cgit v1.2.1