diff options
Diffstat (limited to 'src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp')
| -rw-r--r-- | src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp b/src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp index 81f4645cbc..28c90610de 100644 --- a/src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp +++ b/src/armnn/backends/RefWorkloads/RefDivisionFloat32Workload.cpp @@ -5,7 +5,7 @@ #include "RefDivisionFloat32Workload.hpp" -#include "Division.hpp" +#include "ArithmeticFunction.hpp" #include "RefWorkloadUtils.hpp" #include "Profiling.hpp" @@ -25,7 +25,7 @@ void RefDivisionFloat32Workload::Execute() const const float* inputData0 = GetInputTensorDataFloat(0, m_Data); const float* inputData1 = GetInputTensorDataFloat(1, m_Data); - Division(inShape0, inShape1, outShape, inputData0, inputData1, outputData); + ArithmeticFunction<std::divides<float>>(inShape0, inShape1, outShape, inputData0, inputData1, outputData); } } //namespace armnn |