IVGCVSW-6494 Add CpuAcc Batch MatMul Workload Fp32
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I2def6995f81d33e68f1ea45d8d19a1e6294049b1
diff --git a/src/backends/backendsCommon/WorkloadUtils.cpp b/src/backends/backendsCommon/WorkloadUtils.cpp
index b045530..3aea667 100644
--- a/src/backends/backendsCommon/WorkloadUtils.cpp
+++ b/src/backends/backendsCommon/WorkloadUtils.cpp
@@ -341,4 +341,24 @@
return keyIndices;
}
+armnn::PermutationVector GeneratePermutationVectorOnLastTwoDimensions(unsigned int rank)
+{
+ armnn::PermutationVector permutationVector{};
+ switch (rank)
+ {
+ case 2:
+ permutationVector = {1U, 0U};
+ break;
+ case 3:
+ permutationVector = {0U, 2U, 1U};
+ break;
+ case 4:
+ permutationVector = {0U, 1U, 3U, 2U};
+ break;
+ default:
+ throw Exception("Invalid number of dimensions.");
+ }
+ return permutationVector;
+}
+
} // namespace armnn