aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
diff options
context:
space:
mode:
authorMatthew Bentham <matthew.bentham@arm.com>2018-10-01 11:32:48 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-10 16:16:57 +0100
commite2ec330b015fc34331f0023283b964ef97e1c5bb (patch)
tree76ce917f5d243ceb473dcdc2263e09781954b635 /src/backends/cl/workloads/ClMultiplicationWorkload.hpp
parentb30c53342756699dfaa6effd0a94aa9e69c2063a (diff)
downloadarmnn-e2ec330b015fc34331f0023283b964ef97e1c5bb.tar.gz
IVGCVSW-1207 - Remove typing from ClMultiplicationWorkload
Don't need this now as it uses the compute library validation function, and all of the code for the supported types is identical. Adds Uint8 support to Cl backend, and unit test cases. Change-Id: I35d4edacc1aca241e95d1b19ae525a23d9513c99
Diffstat (limited to 'src/backends/cl/workloads/ClMultiplicationWorkload.hpp')
-rw-r--r--src/backends/cl/workloads/ClMultiplicationWorkload.hpp34
1 files changed, 34 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClMultiplicationWorkload.hpp b/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
new file mode 100644
index 0000000000..0586be96ed
--- /dev/null
+++ b/src/backends/cl/workloads/ClMultiplicationWorkload.hpp
@@ -0,0 +1,34 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <backends/Workload.hpp>
+
+#include <arm_compute/runtime/CL/CLFunctions.h>
+
+namespace armnn
+{
+
+arm_compute::Status ClMultiplicationWorkloadValidate(const TensorInfo& input0,
+ const TensorInfo& input1,
+ const TensorInfo& output);
+
+class ClMultiplicationWorkload : public BaseWorkload<MultiplicationQueueDescriptor>
+{
+public:
+ ClMultiplicationWorkload(const MultiplicationQueueDescriptor& descriptor, const WorkloadInfo& info);
+
+ using BaseWorkload<MultiplicationQueueDescriptor>::BaseWorkload;
+ void Execute() const override;
+
+private:
+ mutable arm_compute::CLPixelWiseMultiplication m_PixelWiseMultiplication;
+};
+
+} //namespace armnn
+
+
+