diff options
Diffstat (limited to 'src/armnnConverter')
| -rw-r--r-- | src/armnnConverter/ArmnnConverter.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnnConverter/ArmnnConverter.cpp b/src/armnnConverter/ArmnnConverter.cpp index f143e6989b..21b89ea6f8 100644 --- a/src/armnnConverter/ArmnnConverter.cpp +++ b/src/armnnConverter/ArmnnConverter.cpp @@ -21,6 +21,7 @@ #endif #include <HeapProfiling.hpp> +#include <armnn/utility/NumericCast.hpp> #include "armnn/utility/StringUtils.hpp" #include <boost/format.hpp> @@ -49,7 +50,7 @@ armnn::TensorShape ParseTensorShape(std::istream& stream) { try { - result.push_back(boost::numeric_cast<unsigned int>(std::stoi((token)))); + result.push_back(armnn::numeric_cast<unsigned int>(std::stoi((token)))); } catch (const std::exception&) { @@ -59,7 +60,7 @@ armnn::TensorShape ParseTensorShape(std::istream& stream) } } - return armnn::TensorShape(boost::numeric_cast<unsigned int>(result.size()), result.data()); + return armnn::TensorShape(armnn::numeric_cast<unsigned int>(result.size()), result.data()); } bool CheckOption(const po::variables_map& vm, |