IVGCVSW-6169 Add GpuAcc Conv3d Workload

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I8b73dccc14ef71cc083896102e24afb2e56e72e2

1 files changed, 41 insertions, 0 deletions

diff --git a/src/backends/cl/workloads/ClConvolution3dWorkload.hpp b/src/backends/cl/workloads/ClConvolution3dWorkload.hpp
new file mode 100644
index 0000000000..93654ecad8
--- /dev/null
+++ b/src/backends/cl/workloads/ClConvolution3dWorkload.hpp
@@ -0,0 +1,41 @@
+//
+// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <backendsCommon/Workload.hpp>
+
+#include <arm_compute/runtime/CL/functions/CLConv3D.h>
+#include <arm_compute/runtime/MemoryManagerOnDemand.h>
+
+#include <memory>
+
+namespace armnn
+{
+
+arm_compute::Status ClConvolution3dWorkloadValidate(const TensorInfo& input,
+                                                    const TensorInfo& output,
+                                                    const Convolution3dDescriptor& descriptor,
+                                                    const TensorInfo& weights,
+                                                    const Optional<TensorInfo>& biases,
+                                                    bool isFastMathEnabled = false,
+                                                    const ActivationDescriptor* activationDescriptor = nullptr);
+
+class ClConvolution3dWorkload : public BaseWorkload<Convolution3dQueueDescriptor>
+{
+public:
+    ClConvolution3dWorkload(const Convolution3dQueueDescriptor& descriptor,
+                            const WorkloadInfo& info,
+                            std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
+                            const arm_compute::CLCompileContext& clCompileContext,
+                            const bool isFastMathEnabled = false);
+    void Execute() const override;
+
+private:
+    mutable arm_compute::CLConv3D m_ConvolutionLayer;
+};
+
+} //namespace armnn
+