diff options
Diffstat (limited to 'src/backends/reference/workloads/Mean.cpp')
-rw-r--r-- | src/backends/reference/workloads/Mean.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/Mean.cpp b/src/backends/reference/workloads/Mean.cpp index f2c0a4fc3f..72080ef042 100644 --- a/src/backends/reference/workloads/Mean.cpp +++ b/src/backends/reference/workloads/Mean.cpp @@ -128,7 +128,7 @@ void Mean(const armnn::TensorInfo& inputInfo, for (unsigned int idx = 0; idx < numResolvedAxis; ++idx) { unsigned int current = inputDims[resolvedAxis[idx]]; - BOOST_ASSERT(boost::numeric_cast<float>(current) < + ARMNN_ASSERT(boost::numeric_cast<float>(current) < (std::numeric_limits<float>::max() / boost::numeric_cast<float>(numElementsInAxis))); numElementsInAxis *= current; } |