diff options
author | Colm Donelan <Colm.Donelan@arm.com> | 2021-05-27 16:43:25 +0100 |
---|---|---|
committer | Colm Donelan <colm.donelan@arm.com> | 2021-05-31 08:28:34 +0000 |
commit | 41e764c0f0984f6f5b890f857033372e4476dd97 (patch) | |
tree | e8d5079fc8930b99b7d7d6945b0c825f0eec2c0b /include/armnn | |
parent | 3982548e928c58b1c5d876d3b14fae452cef37ca (diff) | |
download | armnn-41e764c0f0984f6f5b890f857033372e4476dd97.tar.gz |
IVGCVSW-6059 Fixing PermutationVector.end() to cope with dimensions < 5
* PermutationVector.end() was returning the end of the fixed size array
m_DimMappings rather than the number of mappings set by the constructor.
Signed-off-by: Colm Donelan <Colm.Donelan@arm.com>
Change-Id: Ie218f7922e8c9c35c1dc702e43a5ee2fd1a61ff0
Diffstat (limited to 'include/armnn')
-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 { |