10 #include <boost/assert.hpp> 11 #include <boost/numeric/conversion/cast.hpp> 18 const void* inputData,
20 unsigned int dataTypeSize)
25 BOOST_ASSERT(descriptor.
m_Begin.size() == numDims);
26 BOOST_ASSERT(descriptor.
m_Size.size() == numDims);
28 constexpr
unsigned int maxNumDims = 4;
29 BOOST_ASSERT(numDims <= maxNumDims);
31 std::vector<unsigned int> paddedInput(4);
32 std::vector<unsigned int> paddedBegin(4);
33 std::vector<unsigned int> paddedSize (4);
35 const unsigned int numPaddingDims = maxNumDims - numDims;
36 for (
unsigned int i = 0u; i < maxNumDims; ++i)
38 if (i < numPaddingDims)
46 const unsigned int j = i - numPaddingDims;
47 paddedInput[i] = inputShape[j];
48 paddedBegin[i] = descriptor.
m_Begin[j];
49 paddedSize[i] = descriptor.
m_Size[j];
53 unsigned int dim0 = paddedInput[0];
54 unsigned int dim1 = paddedInput[1];
55 unsigned int dim2 = paddedInput[2];
56 unsigned int dim3 = paddedInput[3];
58 unsigned int begin0 = paddedBegin[0];
59 unsigned int begin1 = paddedBegin[1];
60 unsigned int begin2 = paddedBegin[2];
61 unsigned int begin3 = paddedBegin[3];
63 unsigned int size0 = paddedSize[0];
64 unsigned int size1 = paddedSize[1];
65 unsigned int size2 = paddedSize[2];
66 unsigned int size3 = paddedSize[3];
68 BOOST_ASSERT(begin0 + size0 <= dim0);
69 BOOST_ASSERT(begin1 + size1 <= dim1);
70 BOOST_ASSERT(begin2 + size2 <= dim2);
71 BOOST_ASSERT(begin3 + size3 <= dim3);
73 const unsigned char* input =
reinterpret_cast<const unsigned char*
>(inputData);
74 unsigned char* output =
reinterpret_cast<unsigned char*
>(outputData);
77 for (
unsigned int idx0 = begin0; idx0 < begin0 + size0; ++idx0)
79 for (
unsigned int idx1 = begin1; idx1 < begin1 + size1; ++idx1)
81 for (
unsigned int idx2 = begin2; idx2 < begin2 + size2; ++idx2)
83 for (
unsigned int idx3 = begin3; idx3 < begin3 + size3; ++idx3)
85 const unsigned int inputOffset =
86 (((idx0 * dim1 + idx1) * dim2 + idx2) * dim3 + idx3) * dataTypeSize;
88 ::memcpy(output, input + inputOffset, dataTypeSize);
89 output += dataTypeSize;
void Slice(const TensorInfo &inputInfo, const SliceDescriptor &descriptor, const void *inputData, void *outputData, unsigned int dataTypeSize)
const TensorShape & GetShape() const
std::vector< unsigned int > m_Size
Size of the slice in each dimension.
Copyright (c) 2020 ARM Limited.
void IgnoreUnused(Ts &&...)
std::vector< unsigned int > m_Begin
Beginning indices of the slice in each dimension.
A SliceDescriptor for the SliceLayer.
unsigned int GetNumDimensions() const