diff options
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
-rw-r--r-- | src/armnn/QuantizerVisitor.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp index d6aee6babf..0dc45822b4 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -27,6 +27,9 @@ public: // Functions to quantize the individual layers, overridden from ILayerVisitor void VisitInputLayer(const IConnectableLayer *layer, LayerBindingId id, const char *name = nullptr) override; void VisitAdditionLayer(const IConnectableLayer *layer, const char *name = nullptr) override; + void VisitActivationLayer(const IConnectableLayer *layer, + const ActivationDescriptor& activationDescriptor, + const char *name = nullptr) override; void VisitOutputLayer(const IConnectableLayer *layer, LayerBindingId id, const char *name = nullptr) override; void VisitBatchNormalizationLayer(const IConnectableLayer* layer, const BatchNormalizationDescriptor& desc, |