diff options
Diffstat (limited to 'src/backends/reference/workloads/RefCastWorkload.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefCastWorkload.cpp | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefCastWorkload.cpp b/src/backends/reference/workloads/RefCastWorkload.cpp new file mode 100644 index 0000000000..7080415e5d --- /dev/null +++ b/src/backends/reference/workloads/RefCastWorkload.cpp @@ -0,0 +1,40 @@ +// +// Copyright © 2021 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "RefCastWorkload.hpp" +#include "RefWorkloadUtils.hpp" +#include <armnnUtils/FloatingPointConverter.hpp> +#include <ResolveType.hpp> +#include "Encoders.hpp" +#include "Decoders.hpp" + +namespace +{ + void Cast(armnn::Decoder<float>& in, armnn::Encoder<float>& out, const uint32_t numElements ) + { + for (unsigned int i = 0; i < numElements; i++) + { + out.Set(in.Get()); + ++in; + ++out; + } + } +} + +namespace armnn +{ + + void RefCastWorkload::Execute() const + { + ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefCastWorkload_Execute"); + const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]); + const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]); + + Cast(*MakeDecoder<float>(inputInfo, m_Data.m_Inputs[0]->Map()), + *MakeEncoder<float>(outputInfo, m_Data.m_Outputs[0]->Map()), + inputInfo.GetNumElements()); + } + +} //namespace armnn
\ No newline at end of file |