diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonWorkloadUtils.hpp')
-rw-r--r-- | src/backends/neon/workloads/NeonWorkloadUtils.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/neon/workloads/NeonWorkloadUtils.hpp b/src/backends/neon/workloads/NeonWorkloadUtils.hpp index 00ecb13008..589d2ea017 100644 --- a/src/backends/neon/workloads/NeonWorkloadUtils.hpp +++ b/src/backends/neon/workloads/NeonWorkloadUtils.hpp @@ -113,4 +113,11 @@ inline auto SetNeonSliceData(const std::vector<unsigned int>& m_begin, return std::make_tuple(starts, ends); } +template <typename DataType, typename PayloadType> +DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data) +{ + ITensorHandle* tensorHandle = data.m_Outputs[idx]; + return reinterpret_cast<DataType*>(tensorHandle->Map()); +} + } //namespace armnn |