diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnn/Types.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 9e46d08501..de475ab68f 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -274,7 +274,11 @@ public: SizeType GetSize() const { return m_NumDimMappings; } ConstIterator begin() const { return m_DimMappings.begin(); } - ConstIterator end() const { return m_DimMappings.end(); } + /** + * + * @return pointer one past the end of the number of mapping not the length of m_DimMappings. + */ + ConstIterator end() const { return m_DimMappings.begin() + m_NumDimMappings; } bool IsEqual(const PermutationVector& other) const { |