2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
8 #include "ITensorHandle.hpp"
10 #include <armnn/Tensor.hpp>
12 #include <boost/cast.hpp>
18 template<typename ArrayType, typename Arg>
19 void AssignValues(unsigned int num, unsigned int& idx, const ArrayType& array, Arg& arg)
26 arg = array[(num - 1) - idx];
30 template<typename T, typename ArrayType, typename ...Args>
31 void AssignValues(unsigned int num, unsigned int idx, const ArrayType& array, T& assignee, Args& ... args)
33 AssignValues(num, idx, array, assignee);
35 AssignValues(num, idx, array, args...);
39 template<typename CopyFunc>
40 void CopyTensorContentsGeneric(const ITensorHandle* srcTensor, ITensorHandle* dstTensor, CopyFunc copy)
42 static_assert(MaxNumOfTensorDimensions == 4, "Please update CopyTensorContents");
44 TensorShape srcStrides = srcTensor->GetStrides();
45 const TensorShape& srcShape = srcTensor->GetShape();
46 TensorShape dstStrides = dstTensor->GetStrides();
47 const TensorShape& dstShape = dstTensor->GetShape();
49 size_t srcBatches = 1;
50 size_t srcChannels = 1;
53 AssignValues(srcShape.GetNumDimensions(),0, srcShape,
59 size_t srcBatchStride = 0;
60 size_t srcChannelStride = 0;
61 size_t srcHeightStride = 0;
62 size_t srcWidthStride = 0;
63 AssignValues(srcStrides.GetNumDimensions(),0, srcStrides,
69 size_t dstBatches = 1;
70 size_t dstChannels = 1;
73 AssignValues(dstShape.GetNumDimensions(),0, dstShape,
79 size_t dstBatchStride = 0;
80 size_t dstChannelStride = 0;
81 size_t dstHeightStride = 0;
82 size_t dstWidthStride = 0;
83 AssignValues(dstStrides.GetNumDimensions(),0, dstStrides,
89 auto srcData = static_cast<const uint8_t*>(srcTensor->Map());
90 auto dstData = static_cast<uint8_t*>(dstTensor->Map());
92 size_t copyLength = std::min(srcWidth*srcWidthStride, dstWidth*dstWidthStride);
93 size_t copyHeight = std::min(srcHeight, dstHeight);
94 size_t copyChannels = std::min(srcChannels, dstChannels);
95 size_t copyBatches = std::min(srcBatches, dstBatches);
97 for(unsigned int b=0; b < copyBatches; ++b)
99 auto srcPtrBatch = srcData;
100 auto dstPtrBatch = dstData;
101 for (unsigned int c=0; c< copyChannels; ++c)
103 auto srcPtrChannel = srcData;
104 auto dstPtrChannel = dstData;
105 for (unsigned int h=0; h < copyHeight; ++h)
107 copy(dstData, srcData, copyLength);
108 dstData += dstHeightStride;
109 srcData += srcHeightStride;
111 dstData += (static_cast<long>(dstChannelStride) - (dstData - dstPtrChannel));
112 srcData += (static_cast<long>(srcChannelStride) - (srcData - srcPtrChannel));
114 dstData += (static_cast<long>(dstBatchStride)-(dstData - dstPtrBatch));
115 srcData += (static_cast<long>(srcBatchStride)-(srcData - srcPtrBatch));
122 template <typename SrcTensorHandleType, typename DstTensorHandleType, typename DescriptorType>
123 void GatherTensorHandlePairs(const DescriptorType& descriptor,
124 std::vector<std::pair<SrcTensorHandleType*, DstTensorHandleType*>>& tensorHandlePairs)
126 const unsigned int numInputs = static_cast<unsigned int>(descriptor.m_Inputs.size());
127 tensorHandlePairs.reserve(numInputs);
129 for (unsigned int i = 0; i < numInputs; ++i)
131 SrcTensorHandleType* const srcTensorHandle = boost::polymorphic_downcast<SrcTensorHandleType*>(
132 descriptor.m_Inputs[i]);
133 DstTensorHandleType* const dstTensorHandle = boost::polymorphic_downcast<DstTensorHandleType*>(
134 descriptor.m_Outputs[i]);
136 tensorHandlePairs.emplace_back(srcTensorHandle, dstTensorHandle);