From aa06714b9f49fe36716fd3c116aadd892fadd4ac Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Tue, 11 Jun 2019 16:01:44 +0100 Subject: IVGCVSW-3260 Add serialization support for SpaceToDepth Signed-off-by: Aron Virginas-Tar Change-Id: Ie0d9561437ff5da8fa4db81fee1e70bd18c47034 --- src/armnnSerializer/Serializer.cpp | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'src/armnnSerializer/Serializer.cpp') diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 676d10c0e4..131bb95608 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -788,7 +788,17 @@ void SerializerVisitor::VisitSpaceToDepthLayer(const armnn::IConnectableLayer* l const armnn::SpaceToDepthDescriptor& spaceToDepthDescriptor, const char* name) { - throw armnn::Exception("SerializerVisitor::VisitSpaceToDepthLayer is not yet implemented"); + auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_SpaceToDepth); + auto flatBufferDescriptor = + CreateSpaceToDepthDescriptor(m_flatBufferBuilder, + spaceToDepthDescriptor.m_BlockSize, + GetFlatBufferDataLayout(spaceToDepthDescriptor.m_DataLayout)); + + auto flatBufferLayer = serializer::CreateSpaceToDepthLayer(m_flatBufferBuilder, + flatBufferBaseLayer, + flatBufferDescriptor); + + CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_SpaceToDepthLayer); } // Build FlatBuffer for Splitter Layer -- cgit v1.2.1