diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-01 11:32:48 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
commit | e2ec330b015fc34331f0023283b964ef97e1c5bb (patch) | |
tree | 76ce917f5d243ceb473dcdc2263e09781954b635 /src/backends/cl/workloads/ClMultiplicationWorkload.hpp | |
parent | b30c53342756699dfaa6effd0a94aa9e69c2063a (diff) | |
download | armnn-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.hpp | 34 |
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 + + + |