diff options
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
| -rw-r--r-- | src/armnn/QuantizerVisitor.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp index ab773d42eb..1f1f651c1a 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -79,12 +79,16 @@ public: void VisitInputLayer(const IConnectableLayer* layer, LayerBindingId id, const char* name = nullptr) override; void VisitMeanLayer(const IConnectableLayer* layer, - const MeanDescriptor& meanDescriptor, - const char* name = nullptr) override; + const MeanDescriptor& meanDescriptor, + const char* name = nullptr) override; void VisitMultiplicationLayer(const IConnectableLayer* layer, const char* name = nullptr) override; + void VisitNormalizationLayer(const armnn::IConnectableLayer* layer, + const armnn::NormalizationDescriptor& normalizationDescriptor, + const char* name = nullptr) override; + void VisitOutputLayer(const IConnectableLayer* layer, LayerBindingId id, const char* name = nullptr) override; void VisitPadLayer(const IConnectableLayer*, |