aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefElementwiseWorkload.cpp
diff options
context:
space:
mode:
authorFinn Williams <Finn.Williams@arm.com>2020-06-22 15:58:32 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2020-06-30 13:00:52 +0000
commitcbd2c230b7ce5f26e2ccccf36b7ad450f6e1ad09 (patch)
treeeb6e5393726be21213e72f26a676b7c3809fc995 /src/backends/reference/workloads/RefElementwiseWorkload.cpp
parent532a29d12d72f54549d8b71edd485c17af65698a (diff)
downloadarmnn-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.cpp')
-rw-r--r--src/backends/reference/workloads/RefElementwiseWorkload.cpp24
1 files changed, 24 insertions, 0 deletions
diff --git a/src/backends/reference/workloads/RefElementwiseWorkload.cpp b/src/backends/reference/workloads/RefElementwiseWorkload.cpp
index 18bf0a7ad9..60acbd6252 100644
--- a/src/backends/reference/workloads/RefElementwiseWorkload.cpp
+++ b/src/backends/reference/workloads/RefElementwiseWorkload.cpp
@@ -67,22 +67,46 @@ template class armnn::RefElementwiseWorkload<std::plus<float>,
armnn::AdditionQueueDescriptor,
armnn::StringMapping::RefAdditionWorkload_Execute>;
+template class armnn::RefElementwiseWorkload<std::plus<int32_t>,
+ armnn::AdditionQueueDescriptor,
+ armnn::StringMapping::RefAdditionWorkload_Execute>;
+
template class armnn::RefElementwiseWorkload<std::minus<float>,
armnn::SubtractionQueueDescriptor,
armnn::StringMapping::RefSubtractionWorkload_Execute>;
+template class armnn::RefElementwiseWorkload<std::minus<int32_t>,
+ armnn::SubtractionQueueDescriptor,
+ armnn::StringMapping::RefSubtractionWorkload_Execute>;
+
template class armnn::RefElementwiseWorkload<std::multiplies<float>,
armnn::MultiplicationQueueDescriptor,
armnn::StringMapping::RefMultiplicationWorkload_Execute>;
+template class armnn::RefElementwiseWorkload<std::multiplies<int32_t>,
+ armnn::MultiplicationQueueDescriptor,
+ armnn::StringMapping::RefMultiplicationWorkload_Execute>;
+
template class armnn::RefElementwiseWorkload<std::divides<float>,
armnn::DivisionQueueDescriptor,
armnn::StringMapping::RefDivisionWorkload_Execute>;
+template class armnn::RefElementwiseWorkload<std::divides<int32_t>,
+ armnn::DivisionQueueDescriptor,
+ armnn::StringMapping::RefDivisionWorkload_Execute>;
+
template class armnn::RefElementwiseWorkload<armnn::maximum<float>,
armnn::MaximumQueueDescriptor,
armnn::StringMapping::RefMaximumWorkload_Execute>;
+template class armnn::RefElementwiseWorkload<armnn::maximum<int32_t>,
+ armnn::MaximumQueueDescriptor,
+ armnn::StringMapping::RefMaximumWorkload_Execute>;
+
template class armnn::RefElementwiseWorkload<armnn::minimum<float>,
armnn::MinimumQueueDescriptor,
armnn::StringMapping::RefMinimumWorkload_Execute>;
+
+template class armnn::RefElementwiseWorkload<armnn::minimum<int32_t>,
+ armnn::MinimumQueueDescriptor,
+ armnn::StringMapping::RefMinimumWorkload_Execute>;