diff options
Diffstat (limited to 'src/armnn/Optimizer.cpp')
-rw-r--r-- | src/armnn/Optimizer.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/Optimizer.cpp b/src/armnn/Optimizer.cpp index 1d6a52efed..3d1b67e2cd 100644 --- a/src/armnn/Optimizer.cpp +++ b/src/armnn/Optimizer.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017,2024 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // #include "Optimizer.hpp" @@ -29,7 +29,11 @@ void Optimizer::Pass(Graph& graph, const Optimizations& optimizations) --it; for (auto&& optimization : optimizations) { - ARMNN_ASSERT(*it); + if (!*it) + { + throw armnn::NullPointerException("Layer must not be null."); + } + optimization->Run(graph, **it); if ((*it)->IsOutputUnconnected()) |