diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2021-02-09 15:56:23 +0000 |
|---|---|---|
| committer | Finn Williams <Finn.Williams@arm.com> | 2021-02-12 13:10:20 +0000 |
| commit | b454c5c65efb238c130b042ace390b2bc7f0bf75 (patch) | |
| tree | d6681d0abf416b3cc280bc3bb70e7d55dfd40a0d /src/armnn/NetworkQuantizerUtils.hpp | |
| parent | 8eae955f665f371b0a2c7c1a06e8ba442afa2298 (diff) | |
| download | armnn-b454c5c65efb238c130b042ace390b2bc7f0bf75.tar.gz | |
IVGCVSW-4893 Refactor ILayerVisitor using unified interface strategy.
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: Id7bc8255a8e3f9e5aac65d510bec8a559bf37246
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 |