aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-03-01 16:14:13 +0000
committerNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-03-04 10:14:57 +0000
commit6522cdcd8b07aa8f423f832305eed57d79891e92 (patch)
treec6f757413eb0a904c6bb6a5c1080d927acaeb84d /src/armnnSerializer/Serializer.cpp
parent79ffdf57b12da6ed3fe70e62311af6c4661a9bd3 (diff)
downloadarmnn-6522cdcd8b07aa8f423f832305eed57d79891e92.tar.gz
IVGCVSW-2705 Add Serializer and Deserializer for ResizeBilinear
Change-Id: Ibc5689a2e00d38dc98ef39e50ed5dc3b91791e16 Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp19
1 files changed, 19 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index b55adb266d..d6df61cbcf 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -424,6 +424,25 @@ void SerializerVisitor::VisitReshapeLayer(const armnn::IConnectableLayer* layer,
CreateAnyLayer(flatBufferReshapeLayer.o, serializer::Layer::Layer_ReshapeLayer);
}
+void SerializerVisitor::VisitResizeBilinearLayer(const armnn::IConnectableLayer* layer,
+ const armnn::ResizeBilinearDescriptor& resizeDescriptor,
+ const char* name)
+{
+ auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_ResizeBilinear);
+
+ auto flatBufferDescriptor =
+ CreateResizeBilinearDescriptor(m_flatBufferBuilder,
+ resizeDescriptor.m_TargetWidth,
+ resizeDescriptor.m_TargetHeight,
+ GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout));
+
+ auto flatBufferLayer = serializer::CreateResizeBilinearLayer(m_flatBufferBuilder,
+ flatBufferBaseLayer,
+ flatBufferDescriptor);
+
+ CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ResizeBilinearLayer);
+}
+
void SerializerVisitor::VisitRsqrtLayer(const armnn::IConnectableLayer* layer, const char* name)
{
auto fbRsqrtBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Rsqrt);