diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-07-01 13:51:07 +0100 |
---|---|---|
committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2019-07-01 14:48:57 +0100 |
commit | 970f43b078eba91c66fb64eadbc9803661ffcda8 (patch) | |
tree | 99613d07e62130caff834eadb35f8456b04c63a4 /src/backends/reference/workloads/RefResizeWorkload.cpp | |
parent | 6fb339a7d202a9c64d8c7843d630fe8ab7be9f33 (diff) | |
download | armnn-970f43b078eba91c66fb64eadbc9803661ffcda8.tar.gz |
IVGCVSW-3365 Add reference workload support for ResizeLayer
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Id551690065dca0686ce597d1f0c14fd73163481e
Diffstat (limited to 'src/backends/reference/workloads/RefResizeWorkload.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefResizeWorkload.cpp | 35 |
1 files changed, 35 insertions, 0 deletions
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<Decoder<float>> decoderPtr = MakeDecoder<float>(inputInfo, m_Data.m_Inputs[0]->Map()); + Decoder<float> &decoder = *decoderPtr; + std::unique_ptr<Encoder<float>> encoderPtr = MakeEncoder<float>(outputInfo, m_Data.m_Outputs[0]->Map()); + Encoder<float> &encoder = *encoderPtr; + + Resize(decoder, inputInfo, encoder, outputInfo, m_Data.m_Parameters.m_DataLayout, m_Data.m_Parameters.m_Method); +} + +} //namespace armnn |