diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-25 12:44:53 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-25 12:26:31 +0000 |
| commit | 9926e589d372fa845affd8fce79b5e5934ce3c08 (patch) | |
| tree | ad962db405b97d00a8e756d55c2edb4c7bd5f2b3 /src/backends/reference/workloads/DepthToSpace.cpp | |
| parent | 24e8f9209249c9d41e57748a1b9d2f5f978db4ee (diff) | |
| download | armnn-9926e589d372fa845affd8fce79b5e5934ce3c08.tar.gz | |
IVGCVSW-3908 Fix DepthToSpace reference unit test failures on Android Q
* Use different way of constructing the PermutationVector objects needed
by the DepthToSpace workload
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ieb32c21b7cb4c2e9af22e9e1a0ac2c534be5559d
Diffstat (limited to 'src/backends/reference/workloads/DepthToSpace.cpp')
| -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); |