From f77cab57b3eca1425384d4d5bfe44d76fc7023b9 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Thu, 1 Jun 2023 16:15:13 +0100 Subject: 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 Change-Id: I7bdd11e4936a27cd97ec65fd915e6ccaa1494cff --- .../reference/workloads/RefSpaceToBatchNdWorkload.cpp | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) (limited to 'src/backends/reference/workloads/RefSpaceToBatchNdWorkload.cpp') diff --git a/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.cpp b/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.cpp index 6aa422afdc..d29c2c801e 100644 --- a/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.cpp +++ b/src/backends/reference/workloads/RefSpaceToBatchNdWorkload.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -7,7 +7,6 @@ #include "SpaceToBatchNd.hpp" #include "RefWorkloadUtils.hpp" -#include namespace armnn { @@ -28,12 +27,12 @@ void RefSpaceToBatchNdWorkload::Execute(std::vector inputs, std: ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefSpaceToBatchNdWorkload_Execute"); const TensorInfo& inputInfo = GetTensorInfo(inputs[0]); - std::unique_ptr> decoder = MakeDecoder(inputInfo, inputs[0]->Map()); - const TensorInfo& outputInfo = GetTensorInfo(outputs[0]); - std::unique_ptr> encoder = MakeEncoder(outputInfo, outputs[0]->Map()); - SpaceToBatchNd(inputInfo, outputInfo, m_Data.m_Parameters, *decoder, *encoder); + std::unique_ptr> inputDecoder = MakeDecoder(inputInfo, inputs[0]->Map()); + std::unique_ptr> outputEncoder = MakeEncoder(outputInfo, outputs[0]->Map()); + + SpaceToBatchNd(inputInfo, outputInfo, m_Data.m_Parameters, *inputDecoder, *outputEncoder); } } //namespace armnn -- cgit v1.2.1