diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-06-11 16:01:44 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-06-12 10:13:50 +0000 |
commit | aa06714b9f49fe36716fd3c116aadd892fadd4ac (patch) | |
tree | 9c2e371d8ff914be6203046751ac18364906f418 /src/armnnSerializer/Serializer.cpp | |
parent | 7f6db40e8c02084a5e960e1f377e8e83ed87f6a1 (diff) | |
download | armnn-aa06714b9f49fe36716fd3c116aadd892fadd4ac.tar.gz |
IVGCVSW-3260 Add serialization support for SpaceToDepth
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ie0d9561437ff5da8fa4db81fee1e70bd18c47034
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 12 |
1 files changed, 11 insertions, 1 deletions
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 |