diff options
Diffstat (limited to 'src/armnnUtils/TensorUtils.cpp')
-rw-r--r-- | src/armnnUtils/TensorUtils.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/armnnUtils/TensorUtils.cpp b/src/armnnUtils/TensorUtils.cpp index 535d68adbe..952c76885a 100644 --- a/src/armnnUtils/TensorUtils.cpp +++ b/src/armnnUtils/TensorUtils.cpp @@ -6,8 +6,8 @@ #include <armnnUtils/TensorUtils.hpp> #include <armnn/backends/ITensorHandle.hpp> +#include <armnn/utility/Assert.hpp> -#include <boost/assert.hpp> #include <boost/format.hpp> #include <boost/numeric/conversion/cast.hpp> @@ -114,8 +114,8 @@ unsigned int GetNumElementsBetween(const TensorShape& shape, const unsigned int firstAxisInclusive, const unsigned int lastAxisExclusive) { - BOOST_ASSERT(firstAxisInclusive <= lastAxisExclusive); - BOOST_ASSERT(lastAxisExclusive <= shape.GetNumDimensions()); + ARMNN_ASSERT(firstAxisInclusive <= lastAxisExclusive); + ARMNN_ASSERT(lastAxisExclusive <= shape.GetNumDimensions()); unsigned int count = 1; for (unsigned int i = firstAxisInclusive; i < lastAxisExclusive; i++) { @@ -126,9 +126,9 @@ unsigned int GetNumElementsBetween(const TensorShape& shape, unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis) { - BOOST_ASSERT_MSG(axis < boost::numeric_cast<int>(inputDimension), + ARMNN_ASSERT_MSG(axis < boost::numeric_cast<int>(inputDimension), "Required axis index greater than number of dimensions."); - BOOST_ASSERT_MSG(axis >= -boost::numeric_cast<int>(inputDimension), + ARMNN_ASSERT_MSG(axis >= -boost::numeric_cast<int>(inputDimension), "Required axis index lower than negative of the number of dimensions"); unsigned int uAxis = axis < 0 ? @@ -140,7 +140,7 @@ unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis) unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int axis) { unsigned int numDim = shape.GetNumDimensions(); - BOOST_ASSERT(axis <= numDim - 1); + ARMNN_ASSERT(axis <= numDim - 1); unsigned int count = 1; for (unsigned int i = axis; i < numDim; i++) { |