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 /src/armnnSerializer/Serializer.cpp | |
| 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 'src/armnnSerializer/Serializer.cpp')
| -rw-r--r-- | src/armnnSerializer/Serializer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index d35be6f6d8..faf3d82d24 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -615,9 +615,9 @@ void SerializerVisitor::VisitPermuteLayer(const armnn::IConnectableLayer* layer, auto flatBufferPermuteBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Permute); std::vector<unsigned int> dimMappings; - for (auto& v: permuteDescriptor.m_DimMappings) + for (unsigned int i=0; i<permuteDescriptor.m_DimMappings.GetSize(); ++i) { - dimMappings.push_back(v); + dimMappings.push_back(permuteDescriptor.m_DimMappings[i]); } auto flatBufferPermuteDesc = serializer::CreatePermuteDescriptor(m_flatBufferBuilder, |