From 92e754f0809b27c0b816a8d1153d60432ac0ed64 Mon Sep 17 00:00:00 2001 From: Nina Drozd Date: Thu, 7 Feb 2019 11:09:27 +0000 Subject: IVGCVSW-2608: support static quantization of Activation Change-Id: Ia9afd15d002d4454ec72f219c5cf214704f6ae31 Signed-off-by: Nina Drozd --- src/armnn/QuantizerVisitor.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/armnn/QuantizerVisitor.hpp') 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, -- cgit v1.2.1