From 34757810f8b734f5f59485a542b56934ad4cc5f0 Mon Sep 17 00:00:00 2001 From: David Monahan Date: Wed, 19 Jun 2019 11:47:21 +0100 Subject: IVGCVSW-3235 Add scalar to use as padding value in Reference Pad Signed-off-by: David Monahan Change-Id: If050f318fcb7626bbfae1b8737a1d232a4a5a915 --- include/armnn/Descriptors.hpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'include') diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 2fda8c1d06..0655d42fbd 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -606,11 +606,11 @@ struct MeanDescriptor /// A PadDescriptor for the PadLayer. struct PadDescriptor { - PadDescriptor() + PadDescriptor() : m_padValue(0) {} - PadDescriptor(const std::vector>& padList) - : m_PadList(padList) + PadDescriptor(const std::vector>& padList, const float& padValue = 0) + : m_PadList(padList), m_padValue(padValue) {} /// @brief Specifies the padding for input dimension. @@ -618,6 +618,9 @@ struct PadDescriptor /// Second is the number of values to add after the tensor in the dimension. /// The number of pairs should match the number of dimensions in the input tensor. std::vector> m_PadList; + + /// Optional value to use for padding, defaults to 0 + float m_padValue; }; /// A StridedSliceDescriptor for the StridedSliceLayer. -- cgit v1.2.1