diff options
Diffstat (limited to 'src/backends/reference/workloads/Splitter.cpp')
| -rw-r--r-- | src/backends/reference/workloads/Splitter.cpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/Splitter.cpp b/src/backends/reference/workloads/Splitter.cpp index 695ae8a088..963e3aa6f3 100644 --- a/src/backends/reference/workloads/Splitter.cpp +++ b/src/backends/reference/workloads/Splitter.cpp @@ -1,12 +1,11 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017, 2024 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // #include "RefWorkloadUtils.hpp" #include <armnn/backends/WorkloadData.hpp> #include <armnn/Tensor.hpp> -#include <armnn/utility/Assert.hpp> #include "Splitter.hpp" #include <cmath> @@ -48,7 +47,9 @@ void Split(const SplitterQueueDescriptor& data, //Split view extents are defined by the size of (the corresponding) input tensor. const TensorInfo& outputInfo = GetTensorInfo(outputs[viewIdx]); - ARMNN_ASSERT(outputInfo.GetNumDimensions() == inputInfo.GetNumDimensions()); + ARMNN_THROW_INVALIDARG_MSG_IF_FALSE( + outputInfo.GetNumDimensions() == inputInfo.GetNumDimensions(), + "The number of output dimensions does not match the number of input dimensions."); // Check all dimensions to see if this element is inside the given input view. bool insideView = true; |