diff options
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r-- | tests/InferenceTest.inl | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl index 4a97d17018..88ce3526e2 100644 --- a/tests/InferenceTest.inl +++ b/tests/InferenceTest.inl @@ -5,6 +5,8 @@ #include "InferenceTest.hpp" #include <armnn/utility/Assert.hpp> +#include <armnn/utility/NumericCast.hpp> + #include <boost/numeric/conversion/cast.hpp> #include <boost/program_options.hpp> #include <fmt/format.h> @@ -130,7 +132,7 @@ TestCaseResult ClassifierTestCase<TTestCaseDatabase, TModel>::ProcessResult(cons unsigned int prediction = 0; boost::apply_visitor([&](auto&& value) { - prediction = boost::numeric_cast<unsigned int>( + prediction = armnn::numeric_cast<unsigned int>( std::distance(value.begin(), std::max_element(value.begin(), value.end()))); }, output); |