aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/Mean.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/Mean.cpp')
-rw-r--r--src/backends/reference/workloads/Mean.cpp8
1 files changed, 3 insertions, 5 deletions
diff --git a/src/backends/reference/workloads/Mean.cpp b/src/backends/reference/workloads/Mean.cpp
index e43a4d5312..fe34efe0c7 100644
--- a/src/backends/reference/workloads/Mean.cpp
+++ b/src/backends/reference/workloads/Mean.cpp
@@ -8,8 +8,6 @@
#include <armnn/utility/NumericCast.hpp>
-#include <boost/numeric/conversion/cast.hpp>
-
#include <cmath>
#include <cstddef>
#include <functional>
@@ -130,15 +128,15 @@ void Mean(const armnn::TensorInfo& inputInfo,
for (unsigned int idx = 0; idx < numResolvedAxis; ++idx)
{
unsigned int current = inputDims[resolvedAxis[idx]];
- ARMNN_ASSERT(boost::numeric_cast<float>(current) <
- (std::numeric_limits<float>::max() / boost::numeric_cast<float>(numElementsInAxis)));
+ ARMNN_ASSERT(armnn::numeric_cast<float>(current) <
+ (std::numeric_limits<float>::max() / armnn::numeric_cast<float>(numElementsInAxis)));
numElementsInAxis *= current;
}
if (numElementsInAxis > 0) {
for (unsigned int idx = 0; idx < numOutputs; ++idx)
{
output[idx];
- output.Set(tempSum[idx] / boost::numeric_cast<float>(numElementsInAxis));
+ output.Set(tempSum[idx] / armnn::numeric_cast<float>(numElementsInAxis));
}
}
}