diff options
author | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-06-28 15:07:10 +0100 |
---|---|---|
committer | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-07-01 10:53:47 +0100 |
commit | 6fb339a7d202a9c64d8c7843d630fe8ab7be9f33 (patch) | |
tree | 08f7119b14c042bc009cae4ca90c0c49c5c15387 /src/armnnSerializer/Serializer.cpp | |
parent | a9075df5b704e4f4432bf26027e3ba671d4596f0 (diff) | |
download | armnn-6fb339a7d202a9c64d8c7843d630fe8ab7be9f33.tar.gz |
IVGCVSW-3364 Add serialization support for Resize layer
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>
Change-Id: I3b1af816cefc1760f63324f365de93f899c9c9ee
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 2d5877db63..57d674095b 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -660,7 +660,20 @@ void SerializerVisitor::VisitResizeLayer(const armnn::IConnectableLayer* layer, const armnn::ResizeDescriptor& resizeDescriptor, const char* name) { - throw armnn::Exception("SerializerVisitor::VisitResizeLayer is not yet implemented"); + auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Resize); + + auto flatBufferDescriptor = + CreateResizeDescriptor(m_flatBufferBuilder, + resizeDescriptor.m_TargetHeight, + resizeDescriptor.m_TargetWidth, + GetFlatBufferResizeMethod(resizeDescriptor.m_Method), + GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout)); + + auto flatBufferLayer = serializer::CreateResizeLayer(m_flatBufferBuilder, + flatBufferBaseLayer, + flatBufferDescriptor); + + CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ResizeLayer); } void SerializerVisitor::VisitRsqrtLayer(const armnn::IConnectableLayer* layer, const char* name) |