diff options
author | David Monahan <david.monahan@arm.com> | 2019-06-19 11:47:21 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-06-24 23:09:02 +0000 |
commit | 34757810f8b734f5f59485a542b56934ad4cc5f0 (patch) | |
tree | c9859e1eb90fdf1e3ba4fbeadb3952cac3477277 /include | |
parent | 84da38b0f11ca3db0a439e510514be780f3933ff (diff) | |
download | armnn-34757810f8b734f5f59485a542b56934ad4cc5f0.tar.gz |
IVGCVSW-3235 Add scalar to use as padding value in Reference Pad
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: If050f318fcb7626bbfae1b8737a1d232a4a5a915
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/Descriptors.hpp | 9 |
1 files changed, 6 insertions, 3 deletions
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<std::pair<unsigned int, unsigned int>>& padList) - : m_PadList(padList) + PadDescriptor(const std::vector<std::pair<unsigned int, unsigned int>>& 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<std::pair<unsigned int, unsigned int>> m_PadList; + + /// Optional value to use for padding, defaults to 0 + float m_padValue; }; /// A StridedSliceDescriptor for the StridedSliceLayer. |