diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-06-01 16:15:13 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-06-22 14:12:28 +0000 |
commit | f77cab57b3eca1425384d4d5bfe44d76fc7023b9 (patch) | |
tree | e51066218697f652a0bc40b618ca279a0f7be3f6 /src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp | |
parent | fd5dbe98c780ae7bd390fae536c2dc636e7b61cc (diff) | |
download | armnn-f77cab57b3eca1425384d4d5bfe44d76fc7023b9.tar.gz |
IVGCVSW-7785 Extend support for 3D tensors BATCH_TO_SPACE and SPACE_TO_BATCH in CpuRef
* Both layers were assuming 4D tensors, now 3D is supported too.
* Remove some unnecessary includes
* Add Unit Tests
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I7bdd11e4936a27cd97ec65fd915e6ccaa1494cff
Diffstat (limited to 'src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp b/src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp index ac6aad3eb2..5fb5835b68 100644 --- a/src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp +++ b/src/backends/reference/workloads/RefBatchToSpaceNdWorkload.hpp @@ -1,14 +1,14 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2018-2019,2021-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once #include "RefBaseWorkload.hpp" -#include <armnn/backends/WorkloadData.hpp> -namespace armnn { +namespace armnn +{ class RefBatchToSpaceNdWorkload : public RefBaseWorkload<BatchToSpaceNdQueueDescriptor> { |