diff options
author | Matthew Jackson <matthew.jackson@arm.com> | 2019-08-15 15:14:18 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-08-16 11:43:00 +0000 |
commit | dba634fd6a66a9e033a1925b0b26c80b270bbf21 (patch) | |
tree | c89740a40b9c109582635b7c40b17a16dd6c0649 /include/armnn | |
parent | 11f99b4e72a92051329b23af7ded759463380086 (diff) | |
download | armnn-dba634fd6a66a9e033a1925b0b26c80b270bbf21.tar.gz |
IVGCVSW-3639 Add 5d tensor support
* Increased MaxNumOfTensorDimensions and fixed issues related to its use
* Fixed issues caused by assuming 5d tensors are invalid
* Updated ArmComputeTensorUtils for 5d tensors
* Added 5d tensor unit tests for add, mul, stack and reshape (needed by IVGCVSW-3527)
Signed-off-by: Matthew Jackson <matthew.jackson@arm.com>
Change-Id: I5bcd64942d0d04efcc6c5acb240ad4b88e010743
Diffstat (limited to 'include/armnn')
-rw-r--r-- | include/armnn/Types.hpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/include/armnn/Types.hpp b/include/armnn/Types.hpp index 12897e2af7..6d8ea6f877 100644 --- a/include/armnn/Types.hpp +++ b/include/armnn/Types.hpp @@ -13,7 +13,7 @@ namespace armnn { -constexpr unsigned int MaxNumOfTensorDimensions = 4U; +constexpr unsigned int MaxNumOfTensorDimensions = 5U; /// @enum Status enumeration /// @var Status::Successful @@ -167,7 +167,12 @@ public: bool IsEqual(const PermutationVector& other) const { - return std::equal(begin(), end(), other.begin(), other.end()); + if (m_NumDimMappings != other.m_NumDimMappings) return false; + for (unsigned int i = 0; i < m_NumDimMappings; ++i) + { + if (m_DimMappings[i] != other.m_DimMappings[i]) return false; + } + return true; } bool IsInverse(const PermutationVector& other) const |