1 // 2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "ClFillWorkload.hpp" 7 8 #include "ClWorkloadUtils.hpp" 9 10 #include <cl/ClTensorHandle.hpp> 11 #include <aclCommon/ArmComputeTensorUtils.hpp> 12 #include <arm_compute/core/Types.h> 13 14 namespace armnn 15 { 16 using namespace armcomputetensorutils; 17 ClFillWorkload(const FillQueueDescriptor & descriptor,const WorkloadInfo & info)18ClFillWorkload::ClFillWorkload(const FillQueueDescriptor& descriptor, const WorkloadInfo& info) 19 : BaseWorkload<FillQueueDescriptor>(descriptor, info) 20 { 21 m_Data.ValidateInputsOutputs("ClFillWorkload", 1, 1); 22 23 arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor(); 24 arm_compute::PixelValue pixelValue = GetPixelValue(output, descriptor.m_Parameters.m_Value); 25 26 m_Layer.configure(&output, pixelValue); 27 } 28 Execute() const29void ClFillWorkload::Execute() const 30 { 31 ARMNN_SCOPED_PROFILING_EVENT_CL("ClFillWorkload_Execute"); 32 RunClFunction(m_Layer, CHECK_LOCATION()); 33 } 34 35 } // namespace armnn 36