aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClMergerWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl/workloads/ClMergerWorkload.hpp')
-rw-r--r--src/backends/cl/workloads/ClMergerWorkload.hpp17
1 files changed, 12 insertions, 5 deletions
diff --git a/src/backends/cl/workloads/ClMergerWorkload.hpp b/src/backends/cl/workloads/ClMergerWorkload.hpp
index 948fb58bd1..8189a1b24a 100644
--- a/src/backends/cl/workloads/ClMergerWorkload.hpp
+++ b/src/backends/cl/workloads/ClMergerWorkload.hpp
@@ -7,18 +7,25 @@
#include <backendsCommon/Workload.hpp>
+#include <arm_compute/runtime/CL/CLFunctions.h>
+
namespace armnn
{
+arm_compute::Status ClMergerWorkloadValidate(const std::vector<const TensorInfo*>& inputs,
+ const TensorInfo& output,
+ const MergerDescriptor& descriptor);
+
class ClMergerWorkload : public BaseWorkload<MergerQueueDescriptor>
{
public:
- using BaseWorkload<MergerQueueDescriptor>::BaseWorkload;
+ ClMergerWorkload(const MergerQueueDescriptor& descriptor, const WorkloadInfo& info);
+
+ void Execute() const override;
- void Execute() const override
- {
- // With subtensors, merger is a no-op.
- }
+private:
+ mutable arm_compute::CLConcatenateLayer m_Layer;
+ bool m_Execute;
};
} //namespace armnn