diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-01 12:14:06 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-01 12:14:06 +0000 |
commit | ebb0f9c1dd97b43c0495eab4f2d4414e2fa3d4b1 (patch) | |
tree | df0707bb555150e24ea9090e0b7778d29061ead5 /src/armnnSerializer/Serializer.cpp | |
parent | 57728788f65656e4fa08923d12bee0de34a72fc7 (diff) | |
download | armnn-ebb0f9c1dd97b43c0495eab4f2d4414e2fa3d4b1.tar.gz |
IVGCVSW-2701 Add Serializer and Deserializer for Pad
Change-Id: I71184236f0394518f29944a77d4b934cbde9e53d
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 2000726526..868a36d42e 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -306,6 +306,29 @@ void SerializerVisitor::VisitMultiplicationLayer(const armnn::IConnectableLayer* CreateAnyLayer(flatBufferMultiplicationLayer.o, serializer::Layer::Layer_MultiplicationLayer); } +void SerializerVisitor::VisitPadLayer(const armnn::IConnectableLayer* layer, + const armnn::PadDescriptor& padDescriptor, + const char* name) +{ + auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Pad); + + std::vector<unsigned int> padList; + for (auto& p: padDescriptor.m_PadList) + { + padList.push_back(p.first); + padList.push_back(p.second); + } + + auto flatBufferPadDesc = serializer::CreatePadDescriptor(m_flatBufferBuilder, + m_flatBufferBuilder.CreateVector(padList)); + + auto flatBufferPadLayer = serializer::CreatePadLayer(m_flatBufferBuilder, + flatBufferBaseLayer, + flatBufferPadDesc); + + CreateAnyLayer(flatBufferPadLayer.o, serializer::Layer::Layer_PadLayer); +} + void SerializerVisitor::VisitPermuteLayer(const armnn::IConnectableLayer* layer, const armnn::PermuteDescriptor& permuteDescriptor, const char* name) |