aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/WorkloadUtils.cpp
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2019-12-11 16:59:01 +0000
committerJim Flynn <jim.flynn@arm.com>2019-12-11 16:59:01 +0000
commit356bfec771858ed435874b525fd88da505380103 (patch)
tree328d85fbe343c9ad92e2cef9921520b58a5bffc1 /src/backends/backendsCommon/WorkloadUtils.cpp
parent26052fcf2f8c91f3479c9484354e88e8944d004d (diff)
downloadarmnn-356bfec771858ed435874b525fd88da505380103.tar.gz
IVGCVSW-4254 Patch Strided Slice CTS failures
Change-Id: Idc7e95f20b1fceb2135db4960877671c155b2f4b Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/WorkloadUtils.cpp')
-rw-r--r--src/backends/backendsCommon/WorkloadUtils.cpp14
1 files changed, 14 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadUtils.cpp b/src/backends/backendsCommon/WorkloadUtils.cpp
index 3185ba00d3..3abadd7563 100644
--- a/src/backends/backendsCommon/WorkloadUtils.cpp
+++ b/src/backends/backendsCommon/WorkloadUtils.cpp
@@ -184,4 +184,18 @@ armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstCpuTensorHandle*
return weightPermuted;
}
+int32_t ConvertMaskToACLFormat(int32_t mask, int32_t numDim)
+{
+ int32_t reversedMask = 0;
+ for (unsigned int i = 0; i < boost::numeric_cast<unsigned int>(numDim); ++i)
+ {
+ // Check if bit set in mask for each dimension
+ int32_t bit = (mask & 1 << i) != 0;
+ // Increment the new mask with the bits reversed
+ reversedMask += (bit << std::max(numDim-(boost::numeric_cast<int>(i)+1), 0));
+ }
+
+ return reversedMask;
+}
+
} // namespace armnn