From 1bde8e31935ac375fd0aaa0dfddb0f7851c16087 Mon Sep 17 00:00:00 2001
From: James Conroy <james.conroy@arm.com>
Date: Wed, 22 Jan 2020 16:40:57 +0000
Subject: IVGCVSW-4260 Fix Transpose CpuRef VTS failures

* Adds missing conversion for permutation vector
  from ANN/TF/Numpy format to ArmNN format.
* Corrects legacy tests based on incorrect
  permutation vectors.

Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I767142378055e484d22f3ffba34580c528370cfe
---
 ConversionUtils.hpp | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'ConversionUtils.hpp')

diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index f8622dd6..72a668f7 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -3370,8 +3370,20 @@ bool ConvertTranspose(const HalOperation& operation, const HalModel& model, Conv
 
     std::vector<uint32_t> outputDims(perm.begin(), perm.begin() + rank);
 
+    // Permutation vectors (outputDims) are given in ANN/Tf format, we must convert them to ArmNN format
+    // For ANN/Tf/ACL: output[i] = input[ perm[i] ]
+    // For ArmNN: output[ perm[i] ] = input[i]
+    // e.g. 3,0,1,2 -> 1,2,3,0
+    std::vector<unsigned int> armnnPermuteShape(rank);
+    std::vector<unsigned int>::iterator it;
+    for (unsigned int i = 0u; i < rank; ++i)
+    {
+        it = std::find(outputDims.begin(), outputDims.end(), i);
+        armnnPermuteShape[i] = static_cast<unsigned int>(std::distance(outputDims.begin(), it));
+    }
+
     armnn::PermuteDescriptor permuteDesc;
-    permuteDesc.m_DimMappings = armnn::PermutationVector(outputDims.data(), outputDims.size());
+    permuteDesc.m_DimMappings = armnn::PermutationVector(armnnPermuteShape.data(), armnnPermuteShape.size());
 
     const HalOperand* output = GetOutputOperand<HalPolicy>(operation, 0, model);
     if (!output)
-- 
cgit v1.2.1