IVGCVSW-3235 Add scalar to use as padding value in Reference Pad

Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: If050f318fcb7626bbfae1b8737a1d232a4a5a915

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;