diff options
Diffstat (limited to 'include/armnn/Descriptors.hpp')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index 5f9df6b2c3..8940e0b003 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -347,4 +347,14 @@ struct MeanDescriptor bool m_KeepDims; }; +struct PadDescriptor +{ + PadDescriptor(const std::vector<std::pair<unsigned int, unsigned int>>& padList); + + // first is number of values to add before the tensor in the dimension, + // 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; +}; + } |