diff options
Diffstat (limited to 'src/armnn/NetworkQuantizerUtils.hpp')
-rw-r--r-- | src/armnn/NetworkQuantizerUtils.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/NetworkQuantizerUtils.hpp b/src/armnn/NetworkQuantizerUtils.hpp index 26f67f95b2..6176a9c794 100644 --- a/src/armnn/NetworkQuantizerUtils.hpp +++ b/src/armnn/NetworkQuantizerUtils.hpp @@ -49,10 +49,12 @@ ConstTensor CreateQuantizedConst(const ConstTensor& tensor, std::vector<uint8_t> template <typename LayerContainer> void VisitLayers(const LayerContainer& layerContainer, ILayerVisitor& visitor) { + visitor.StartVisit(); for (auto layer : layerContainer) { layer->Accept(visitor); } + visitor.FinishVisit(); } } // namespace armnn |