aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
authorFinnWilliamsArm <Finn.Williams@arm.com>2019-06-28 15:07:10 +0100
committerFinnWilliamsArm <Finn.Williams@arm.com>2019-07-01 10:53:47 +0100
commit6fb339a7d202a9c64d8c7843d630fe8ab7be9f33 (patch)
tree08f7119b14c042bc009cae4ca90c0c49c5c15387 /src/armnnSerializer/Serializer.cpp
parenta9075df5b704e4f4432bf26027e3ba671d4596f0 (diff)
downloadarmnn-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.cpp15
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)