1 // 2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "RefFloorWorkload.hpp" 7 8 #include "Decoders.hpp" 9 #include "Encoders.hpp" 10 #include "RefWorkloadUtils.hpp" 11 #include "Profiling.hpp" 12 13 namespace armnn 14 { 15 Execute() const16void RefFloorWorkload::Execute() const 17 { 18 Execute(m_Data.m_Inputs, m_Data.m_Outputs); 19 } 20 ExecuteAsync(ExecutionData & executionData)21void RefFloorWorkload::ExecuteAsync(ExecutionData& executionData) 22 { 23 WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data); 24 Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs); 25 } 26 Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const27void RefFloorWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const 28 { 29 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefFloorFloat32Workload_Execute"); 30 31 const TensorInfo &inputTensorInfo = GetTensorInfo(inputs[0]); 32 std::unique_ptr<Decoder<float>> decoderPtr = MakeDecoder<float>(inputTensorInfo, inputs[0]->Map()); 33 Decoder<float> &decoder = *decoderPtr; 34 35 const TensorInfo &outputTensorInfo = GetTensorInfo(outputs[0]); 36 std::unique_ptr<Encoder<float>> encoderPtr = MakeEncoder<float>(outputTensorInfo, outputs[0]->Map()); 37 Encoder<float> &encoder = *encoderPtr; 38 39 unsigned int numElements = GetTensorInfo(inputs[0]).GetNumElements(); 40 41 for (unsigned int i = 0; i < numElements; ++i) 42 { 43 encoder.Set(floorf(decoder.Get())); 44 ++decoder; 45 ++encoder; 46 } 47 } 48 49 } //namespace armnn 50