diff options
author | David Beck <david.beck@arm.com> | 2018-09-11 16:37:14 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-01 14:56:47 +0100 |
commit | 591cdb7ad9176d43163b340542b0fff470a198ea (patch) | |
tree | 67b3a910f85a4ae3850fea34c893525c100feae1 /src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp | |
parent | 549ae37b51facbdf940bff62b45b3b74c1bc63c9 (diff) | |
download | armnn-591cdb7ad9176d43163b340542b0fff470a198ea.tar.gz |
IVGCVSW-1843 : replacing trivial arithmetic helpers
Change-Id: Iddf637694f1a3a7ef00f006a41b8044a35c7e73c
Diffstat (limited to 'src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp')
-rw-r--r-- | src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp b/src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp index a6ed770c40..d10d874137 100644 --- a/src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp +++ b/src/armnn/backends/RefWorkloads/RefDivisionUint8Workload.cpp @@ -5,7 +5,7 @@ #include "RefDivisionUint8Workload.hpp" -#include "Division.hpp" +#include "ArithmeticFunction.hpp" #include "RefWorkloadUtils.hpp" #include "Profiling.hpp" @@ -27,9 +27,13 @@ void RefDivisionUint8Workload::Execute() const auto dequant1 = Dequantize(GetInputTensorDataU8(1, m_Data), inputInfo1); std::vector<float> results(outputInfo.GetNumElements()); - Division( - inputInfo0.GetShape(), inputInfo1.GetShape(), outputInfo.GetShape(), - dequant0.data(), dequant1.data(),results.data()); + + ArithmeticFunction<std::divides<float>>(inputInfo0.GetShape(), + inputInfo1.GetShape(), + outputInfo.GetShape(), + dequant0.data(), + dequant1.data(), + results.data()); Quantize(GetOutputTensorDataU8(0, m_Data), results.data(), outputInfo); } |