diff options
Diffstat (limited to 'src/armnn/layers/UnmapLayer.cpp')
| -rw-r--r-- | src/armnn/layers/UnmapLayer.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/layers/UnmapLayer.cpp b/src/armnn/layers/UnmapLayer.cpp index cfbde211ba..a62536f3bc 100644 --- a/src/armnn/layers/UnmapLayer.cpp +++ b/src/armnn/layers/UnmapLayer.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2020,2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include "UnmapLayer.hpp" @@ -38,7 +38,11 @@ void UnmapLayer::ValidateTensorShapesFromInputs() { // validates that the input is connected. VerifyLayerConnections(1, CHECK_LOCATION()); - ARMNN_ASSERT(GetNumOutputSlots() == 0); + if (GetNumOutputSlots() != 0) + { + throw armnn::LayerValidationException("Output slots must be \"0\" - currently \"" + + std::to_string(GetNumOutputSlots()) + "\"."); + } } void UnmapLayer::ExecuteStrategy(IStrategy& strategy) const |