aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp23
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)