diff options
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 1 | ||||
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 3 | ||||
-rw-r--r-- | src/armnnSerializer/test/SerializerTests.cpp | 1 |
3 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 83275ca248..7969d10598 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -409,6 +409,7 @@ table PadLayer { table PadDescriptor { padList:[uint]; + padValue:float = 0; } table RsqrtLayer { diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 126247bb8c..208262b699 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -577,7 +577,8 @@ void SerializerVisitor::VisitPadLayer(const armnn::IConnectableLayer* layer, } auto flatBufferPadDesc = serializer::CreatePadDescriptor(m_flatBufferBuilder, - m_flatBufferBuilder.CreateVector(padList)); + m_flatBufferBuilder.CreateVector(padList), + padDescriptor.m_padValue); auto flatBufferPadLayer = serializer::CreatePadLayer(m_flatBufferBuilder, flatBufferBaseLayer, diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp index 8b1594e141..294adec12e 100644 --- a/src/armnnSerializer/test/SerializerTests.cpp +++ b/src/armnnSerializer/test/SerializerTests.cpp @@ -1712,6 +1712,7 @@ private: void VerifyDescriptor(const armnn::PadDescriptor& descriptor) { BOOST_TEST(descriptor.m_PadList == m_Descriptor.m_PadList); + BOOST_TEST(descriptor.m_padValue == m_Descriptor.m_padValue); } armnn::PadDescriptor m_Descriptor; |