diff options
Diffstat (limited to 'src/armnn/NetworkQuantizerUtils.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizerUtils.hpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/armnn/NetworkQuantizerUtils.hpp b/src/armnn/NetworkQuantizerUtils.hpp index dd274f9e35..5497e1b898 100644 --- a/src/armnn/NetworkQuantizerUtils.hpp +++ b/src/armnn/NetworkQuantizerUtils.hpp @@ -10,6 +10,7 @@ #include <armnn/Tensor.hpp> #include <armnn/TypesUtils.hpp> #include <armnn/ILayerVisitor.hpp> +#include <armnn/IStrategy.hpp> #include <armnn/utility/Assert.hpp> #include <utility> @@ -56,4 +57,14 @@ void VisitLayers(const LayerContainer& layerContainer, ILayerVisitor& visitor) visitor.FinishVisit(); } +template <typename LayerContainer> +void ApplyStrategyToLayers(const LayerContainer& layerContainer, IStrategy& strategy) +{ + for (auto layer : layerContainer) + { + layer->ExecuteStrategy(strategy); + } + strategy.FinishStrategy(); +} + } // namespace armnn |