diff options
Diffstat (limited to 'src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp')
-rw-r--r-- | src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp b/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp index 2999be9240..116a5f14cb 100644 --- a/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp +++ b/src/armnn/backends/RefWorkloads/RefAdditionUint8Workload.cpp @@ -5,7 +5,7 @@ #include "RefAdditionUint8Workload.hpp" -#include "Addition.hpp" +#include "ArithmeticFunction.hpp" #include "RefWorkloadUtils.hpp" #include "Profiling.hpp" @@ -28,12 +28,12 @@ void RefAdditionUint8Workload::Execute() const std::vector<float> results(outputInfo.GetNumElements()); - Addition(inputInfo0.GetShape(), - inputInfo1.GetShape(), - outputInfo.GetShape(), - dequant0.data(), - dequant1.data(), - results.data()); + ArithmeticFunction<std::plus<float>>(inputInfo0.GetShape(), + inputInfo1.GetShape(), + outputInfo.GetShape(), + dequant0.data(), + dequant1.data(), + results.data()); Quantize(GetOutputTensorDataU8(0, m_Data), results.data(), outputInfo); } |