From 970f43b078eba91c66fb64eadbc9803661ffcda8 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Mon, 1 Jul 2019 13:51:07 +0100 Subject: IVGCVSW-3365 Add reference workload support for ResizeLayer Signed-off-by: Teresa Charlin Change-Id: Id551690065dca0686ce597d1f0c14fd73163481e --- .../reference/workloads/RefResizeWorkload.cpp | 35 ++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 src/backends/reference/workloads/RefResizeWorkload.cpp (limited to 'src/backends/reference/workloads/RefResizeWorkload.cpp') diff --git a/src/backends/reference/workloads/RefResizeWorkload.cpp b/src/backends/reference/workloads/RefResizeWorkload.cpp new file mode 100644 index 0000000000..26225f8823 --- /dev/null +++ b/src/backends/reference/workloads/RefResizeWorkload.cpp @@ -0,0 +1,35 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "RefResizeWorkload.hpp" + +#include "RefWorkloadUtils.hpp" +#include "Resize.hpp" +#include "BaseIterator.hpp" +#include "Profiling.hpp" + +#include "BaseIterator.hpp" +#include "Decoders.hpp" +#include "Encoders.hpp" + +namespace armnn +{ + +void RefResizeWorkload::Execute() const +{ + ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefResizeWorkload_Execute"); + + const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); + const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]); + + std::unique_ptr> decoderPtr = MakeDecoder(inputInfo, m_Data.m_Inputs[0]->Map()); + Decoder &decoder = *decoderPtr; + std::unique_ptr> encoderPtr = MakeEncoder(outputInfo, m_Data.m_Outputs[0]->Map()); + Encoder &encoder = *encoderPtr; + + Resize(decoder, inputInfo, encoder, outputInfo, m_Data.m_Parameters.m_DataLayout, m_Data.m_Parameters.m_Method); +} + +} //namespace armnn -- cgit v1.2.1