2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
6 #include "RefConvertFp32ToFp16Workload.hpp"
8 #include "FloatingPointConverter.hpp"
9 #include "RefWorkloadUtils.hpp"
10 #include "Profiling.hpp"
17 void RefConvertFp32ToFp16Workload::Execute() const
19 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefConvertFp32ToFp16Workload_Execute");
21 const float* const input = GetInputTensorDataFloat(0, m_Data);
22 Half* const output = GetOutputTensorDataHalf(0, m_Data);
24 // convert Fp32 input to Fp16 output
25 unsigned int numElements = GetTensorInfo(m_Data.m_Inputs[0]).GetNumElements();
26 armnnUtils::FloatingPointConverter::ConvertFloat32To16(input, numElements, output);