diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-06-22 15:58:32 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-06-30 13:00:52 +0000 |
commit | cbd2c230b7ce5f26e2ccccf36b7ad450f6e1ad09 (patch) | |
tree | eb6e5393726be21213e72f26a676b7c3809fc995 /src/backends/reference/workloads/RefElementwiseWorkload.hpp | |
parent | 532a29d12d72f54549d8b71edd485c17af65698a (diff) | |
download | armnn-cbd2c230b7ce5f26e2ccccf36b7ad450f6e1ad09.tar.gz |
IVGCVSW-5007 Implement an Int32 reference Elementwise workload
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I6592169b74ac4294bc09647879aec0718c641f91
Diffstat (limited to 'src/backends/reference/workloads/RefElementwiseWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefElementwiseWorkload.hpp | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseWorkload.hpp b/src/backends/reference/workloads/RefElementwiseWorkload.hpp index 264ddce2de..03683b1a06 100644 --- a/src/backends/reference/workloads/RefElementwiseWorkload.hpp +++ b/src/backends/reference/workloads/RefElementwiseWorkload.hpp @@ -35,33 +35,39 @@ private: std::unique_ptr<Encoder<OutType>> m_Output; }; +template <typename DataType = float> using RefAdditionWorkload = - RefElementwiseWorkload<std::plus<float>, + RefElementwiseWorkload<std::plus<DataType>, AdditionQueueDescriptor, StringMapping::RefAdditionWorkload_Execute>; +template <typename DataType = float> using RefSubtractionWorkload = - RefElementwiseWorkload<std::minus<float>, + RefElementwiseWorkload<std::minus<DataType>, SubtractionQueueDescriptor, StringMapping::RefSubtractionWorkload_Execute>; +template <typename DataType = float> using RefMultiplicationWorkload = - RefElementwiseWorkload<std::multiplies<float>, + RefElementwiseWorkload<std::multiplies<DataType>, MultiplicationQueueDescriptor, StringMapping::RefMultiplicationWorkload_Execute>; +template <typename DataType = float> using RefDivisionWorkload = - RefElementwiseWorkload<std::divides<float>, + RefElementwiseWorkload<std::divides<DataType>, DivisionQueueDescriptor, StringMapping::RefDivisionWorkload_Execute>; +template <typename DataType = float> using RefMaximumWorkload = - RefElementwiseWorkload<armnn::maximum<float>, + RefElementwiseWorkload<armnn::maximum<DataType>, MaximumQueueDescriptor, StringMapping::RefMaximumWorkload_Execute>; +template <typename DataType = float> using RefMinimumWorkload = - RefElementwiseWorkload<armnn::minimum<float>, + RefElementwiseWorkload<armnn::minimum<DataType>, MinimumQueueDescriptor, StringMapping::RefMinimumWorkload_Execute>; |