diff options
Diffstat (limited to 'src/backends/neon/NeonTensorHandle.hpp')
-rw-r--r-- | src/backends/neon/NeonTensorHandle.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/neon/NeonTensorHandle.hpp b/src/backends/neon/NeonTensorHandle.hpp index 11d20878d7..fb2c2b5128 100644 --- a/src/backends/neon/NeonTensorHandle.hpp +++ b/src/backends/neon/NeonTensorHandle.hpp @@ -7,6 +7,8 @@ #include <BFloat16.hpp> #include <Half.hpp> +#include <armnn/utility/Assert.hpp> + #include <aclCommon/ArmComputeTensorHandle.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> @@ -61,7 +63,7 @@ public: // If we have enabled Importing, don't manage the tensor if (!m_IsImportEnabled) { - BOOST_ASSERT(m_MemoryGroup != nullptr); + ARMNN_ASSERT(m_MemoryGroup != nullptr); m_MemoryGroup->manage(&m_Tensor); } } |