diff options
author | keidav01 <keith.davis@arm.com> | 2018-12-10 18:16:07 +0000 |
---|---|---|
committer | Keith Davis Arm <keith.davis@arm.com> | 2018-12-18 10:47:58 +0000 |
commit | d74dc91af2d1302bf9024fcb1690d5df035f9c15 (patch) | |
tree | 0da683c1beb5cfc0c4264bc1d00bac4bf462f73a /src/backends/cl/workloads/ClWorkloadUtils.hpp | |
parent | 2099595bac339c953bf4291be12703c8845e916e (diff) | |
download | armnn-d74dc91af2d1302bf9024fcb1690d5df035f9c15.tar.gz |
IVGCVSW-2131 Implementation of StridedSlice operation
* Added ACl Implementation to ArmNN
* Added helper function for setting CL Strided Slice arguments
Change-Id: Ie10d387fdb054027ea9b8782743d270bb72949c1
Diffstat (limited to 'src/backends/cl/workloads/ClWorkloadUtils.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClWorkloadUtils.hpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClWorkloadUtils.hpp b/src/backends/cl/workloads/ClWorkloadUtils.hpp index 3bee2427f0..32dacdfc61 100644 --- a/src/backends/cl/workloads/ClWorkloadUtils.hpp +++ b/src/backends/cl/workloads/ClWorkloadUtils.hpp @@ -39,6 +39,27 @@ void CopyArmComputeClTensorData(arm_compute::CLTensor& dstTensor, const T* srcDa dstTensor.unmap(); } +inline auto SetClStridedSliceData(const std::vector<int>& m_begin, + const std::vector<int>& m_end, + const std::vector<int>& m_stride) +{ + arm_compute::Coordinates starts; + arm_compute::Coordinates ends; + arm_compute::Coordinates strides; + + unsigned int num_dims = static_cast<unsigned int>(m_begin.size()); + + for (unsigned int i = 0; i < num_dims; i++) { + unsigned int revertedIndex = num_dims - i - 1; + + starts.set(i, static_cast<int>(m_begin[revertedIndex])); + ends.set(i, static_cast<int>(m_end[revertedIndex])); + strides.set(i, static_cast<int>(m_stride[revertedIndex])); + } + + return std::make_tuple(starts, ends, strides); +} + inline void InitializeArmComputeClTensorData(arm_compute::CLTensor& clTensor, const ConstCpuTensorHandle* handle) { |