MCPcopy Create free account
hub / github.com/ARM-software/armnn / Unroll

Method Unroll

src/armnnUtils/Transpose.cpp:48–70  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

46 }
47
48 void Unroll(const void* srcData, void* dstData, size_t dataTypeSize)
49 {
50 if (srcData == nullptr)
51 {
52 throw armnn::Exception("Transpose: Source Data pointer is null");
53 }
54 if (dstData == nullptr)
55 {
56 throw armnn::Exception("Transpose: Destination Data pointer is null");
57 }
58 if (dataTypeSize == 0)
59 {
60 throw armnn::Exception("Transpose: dataTypeSize is zero");
61 }
62
63 const unsigned char* srcDataPtr = reinterpret_cast<const unsigned char*>(srcData);
64 unsigned char* dstDataPtr = reinterpret_cast<unsigned char*>(dstData);
65
66 const unsigned char* const srcEndPtr = srcDataPtr + m_SrcShape.GetNumElements() * dataTypeSize;
67 unsigned char* const dstEndPtr = dstDataPtr + m_SrcShape.GetNumElements() * dataTypeSize;
68
69 Unroll(0, srcDataPtr, dstDataPtr, srcEndPtr, dstEndPtr, dataTypeSize);
70 }
71
72private:
73 void Unroll(size_type dimension,

Callers 1

TransposeFunction · 0.45

Calls 3

ExceptionClass · 0.50
GetNumElementsMethod · 0.45
GetNumDimensionsMethod · 0.45

Tested by

no test coverage detected