diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/backends/reference/workloads/DepthToSpace.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/DepthToSpace.cpp b/src/backends/reference/workloads/DepthToSpace.cpp index 046bd47a6f..d500e9b100 100644 --- a/src/backends/reference/workloads/DepthToSpace.cpp +++ b/src/backends/reference/workloads/DepthToSpace.cpp @@ -50,7 +50,7 @@ void DepthToSpace(const TensorInfo& inputInfo, // batch separately and execute 5D permutations TensorShape permDestShape; - std::initializer_list<unsigned int> permVector; + PermutationVector permVector{}; if (descriptor.m_DataLayout == DataLayout::NCHW) { permDestShape = TensorShape({ outDepth, inHeight, blockSize, inWidth, blockSize }); @@ -69,7 +69,7 @@ void DepthToSpace(const TensorInfo& inputInfo, const uintptr_t batchDataOffset = batchIndex * (numElementsPerBatch * dataTypeSize); armnnUtils::Permute(permDestShape, - PermutationVector(permVector), + permVector, static_cast<const void*>(reinterpret_cast<const uint8_t*>(inputData) + batchDataOffset), static_cast<void*>(reinterpret_cast<uint8_t*>(outputData) + batchDataOffset), dataTypeSize); |