diff options
Diffstat (limited to 'include/armnn')
-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. |