aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClFillWorkload.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl/workloads/ClFillWorkload.cpp')
-rw-r--r--src/backends/cl/workloads/ClFillWorkload.cpp35
1 files changed, 35 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClFillWorkload.cpp b/src/backends/cl/workloads/ClFillWorkload.cpp
new file mode 100644
index 0000000000..47a70bc677
--- /dev/null
+++ b/src/backends/cl/workloads/ClFillWorkload.cpp
@@ -0,0 +1,35 @@
+//
+// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "ClFillWorkload.hpp"
+
+#include "ClWorkloadUtils.hpp"
+
+#include <cl/ClTensorHandle.hpp>
+#include <aclCommon/ArmComputeTensorUtils.hpp>
+#include <arm_compute/core/Types.h>
+
+namespace armnn
+{
+using namespace armcomputetensorutils;
+
+ClFillWorkload::ClFillWorkload(const FillQueueDescriptor& descriptor, const WorkloadInfo& info)
+ : BaseWorkload<FillQueueDescriptor>(descriptor, info)
+{
+ m_Data.ValidateInputsOutputs("ClFillWorkload", 1, 1);
+
+ arm_compute::ICLTensor& output = static_cast<IClTensorHandle*>(this->m_Data.m_Outputs[0])->GetTensor();
+ arm_compute::PixelValue pixelValue = GetPixelValue(output, descriptor.m_Parameters.m_Value);
+
+ m_Layer.configure(&output, pixelValue);
+}
+
+void ClFillWorkload::Execute() const
+{
+ ARMNN_SCOPED_PROFILING_EVENT_CL("ClFillWorkload_Execute");
+ RunClFunction(m_Layer, CHECK_LOCATION());
+}
+
+} // namespace armnn