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,const arm_compute::CLCompileContext & clCompileContext)18 ClFillWorkload::ClFillWorkload(const FillQueueDescriptor& descriptor,
19 const WorkloadInfo& info,
20 const arm_compute::CLCompileContext& clCompileContext)
21 : ClBaseWorkload<FillQueueDescriptor>(descriptor, info)
22 {
23 // Report Profiling Details
24 ARMNN_REPORT_PROFILING_WORKLOAD_DESC("ClFillWorkload_Construct",
25 descriptor.m_Parameters,
26 info,
27 this->GetGuid());
28
29 m_Data.ValidateInputsOutputs("ClFillWorkload", 1, 1);
30
31 arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor();
32 arm_compute::PixelValue pixelValue = GetPixelValue(output.info(), descriptor.m_Parameters.m_Value);
33
34 {
35 ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "ClFillWorkload_configure");
36 m_Layer.configure(clCompileContext, &output, pixelValue);
37 }
38 }
39
Execute() const40 void ClFillWorkload::Execute() const
41 {
42 ARMNN_SCOPED_PROFILING_EVENT_CL_GUID("ClFillWorkload_Execute", this->GetGuid());
43 RunClFunction(m_Layer, CHECK_LOCATION());
44 }
45
46 } // namespace armnn
47