19 #include <boost/cast.hpp> 26 template <
typename ArrayType,
typename Arg>
27 void AssignValues(
unsigned int num,
unsigned int& idx,
const ArrayType& array, Arg& arg)
34 arg = array[(num - 1) - idx];
38 template <
typename T,
typename ArrayType,
typename... Args>
39 void AssignValues(
unsigned int num,
unsigned int idx,
const ArrayType& array, T& assignee, Args&... args)
41 AssignValues(num, idx, array, assignee);
43 AssignValues(num, idx, array, args...);
48 template <
typename CopyFunc>
57 const auto srcSize = srcTensor->
GetStrides()[0] * srcShape[0];
58 boost::ignore_unused(srcSize);
61 const auto dstSize = dstTensor->
GetStrides()[0] * dstShape[0];
62 boost::ignore_unused(dstSize);
65 size_t srcBatches = 1;
68 size_t srcChannels = 1;
78 size_t srcDepthStride = 0;
79 size_t srcBatchStride = 0;
80 size_t srcHeightStride = 0;
81 size_t srcWidthStride = 0;
82 size_t srcChannelStride = 0;
93 size_t dstBatches = 1;
96 size_t dstChannels = 1;
106 size_t dstDepthStride = 0;
107 size_t dstBatchStride = 0;
108 size_t dstHeightStride = 0;
109 size_t dstWidthStride = 0;
110 size_t dstChannelStride = 0;
120 const unsigned char* srcDataStart;
121 unsigned char* dstDataStart;
124 srcDataStart =
static_cast<const uint8_t*
>(srcTensor->
Map());
125 dstDataStart =
static_cast<uint8_t*
>(dstTensor->
Map());
128 size_t copyLength = std::min(srcChannels * srcChannelStride, dstChannels * dstChannelStride);
129 size_t copyWidth = std::min(srcWidth, dstWidth);
130 size_t copyHeight = std::min(srcHeight, dstHeight);
131 size_t copyBatches = std::min(srcBatches, dstBatches);
132 size_t copyDepth = std::min(srcDepth, dstDepth);
137 if (copyLength == srcWidthStride &&
138 copyLength == dstWidthStride)
142 copyLength *= copyWidth;
145 if (copyLength == srcHeightStride &&
146 copyLength == dstHeightStride)
150 copyLength *= copyHeight;
155 const unsigned char* srcData = srcDataStart;
156 unsigned char* dstData = dstDataStart;
157 for (
unsigned int d = 0; d < copyDepth; ++d)
159 auto srcPtrDepth = srcData;
160 auto dstPtrDepth = dstData;
161 for (
unsigned int b = 0; b < copyBatches; ++b)
163 auto srcPtrBatch = srcData;
164 auto dstPtrBatch = dstData;
165 for (
unsigned int h = 0; h < copyHeight; ++h)
167 auto srcPtrChannel = srcData;
168 auto dstPtrChannel = dstData;
169 for (
unsigned int w = 0; w < copyWidth; ++w)
171 BOOST_ASSERT(srcData >= srcDataStart && srcData + copyLength <= srcDataStart + srcSize);
172 BOOST_ASSERT(dstData >= dstDataStart && dstData + copyLength <= dstDataStart + dstSize);
173 copy(dstData, srcData, copyLength);
174 dstData += dstWidthStride;
175 srcData += srcWidthStride;
177 dstData += (
static_cast<long>(dstHeightStride) - (dstData - dstPtrChannel));
178 srcData += (
static_cast<long>(srcHeightStride) - (srcData - srcPtrChannel));
180 dstData += (
static_cast<long>(dstBatchStride) - (dstData - dstPtrBatch));
181 srcData += (
static_cast<long>(srcBatchStride) - (srcData - srcPtrBatch));
183 dstData += (
static_cast<long>(dstDepthStride) - (dstData - dstPtrDepth));
184 srcData += (
static_cast<long>(srcDepthStride) - (srcData - srcPtrDepth));
191 template <
typename SrcTensorHandleType,
typename DstTensorHandleType,
typename DescriptorType>
193 std::vector<std::pair<SrcTensorHandleType*, DstTensorHandleType*>>& tensorHandlePairs)
195 const unsigned int numInputs =
static_cast<unsigned int>(descriptor.m_Inputs.size());
196 tensorHandlePairs.reserve(numInputs);
198 for (
unsigned int i = 0; i < numInputs; ++i)
200 SrcTensorHandleType*
const srcTensorHandle =
201 boost::polymorphic_downcast<SrcTensorHandleType*>(descriptor.m_Inputs[i]);
202 DstTensorHandleType*
const dstTensorHandle =
203 boost::polymorphic_downcast<DstTensorHandleType*>(descriptor.m_Outputs[i]);
205 tensorHandlePairs.emplace_back(srcTensorHandle, dstTensorHandle);
213 void* permuteBuffer);
221 void* permuteBuffer);
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
armnn::ConstTensor PermuteTensor(const ConstCpuTensorHandle *tensor, const PermutationVector &permutationVector, void *permuteBuffer)
unsigned int GetNumDimensions() const
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstCpuTensorHandle *weightTensor, DataLayout dataLayout, void *permuteBuffer)
void CopyTensorContentsGeneric(const ITensorHandle *srcTensor, ITensorHandle *dstTensor, CopyFunc copy)
virtual const void * Map(bool blocking=true) const =0
virtual TensorShape GetShape() const =0
virtual TensorShape GetStrides() const =0
void GatherTensorHandlePairs(const DescriptorType &descriptor, std::vector< std::pair< SrcTensorHandleType *, DstTensorHandleType *>> &tensorHandlePairs)
virtual void Unmap() const =0
Unmap the tensor data.
int32_t ConvertMaskToACLFormat(int32_t mask, int32_t numDim)
constexpr unsigned int MaxNumOfTensorDimensions
void ReshapeWeightsForAcl(TensorInfo &weightInfo, DataLayout dataLayout)
TensorInfo ConvertWeightTensorInfoFromArmnnToAcl(const TensorInfo &weightInfo, DataLayout dataLayout)