aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/QuantizerVisitor.hpp
diff options
context:
space:
mode:
authorFinnWilliamsArm <Finn.Williams@arm.com>2019-09-06 10:04:08 +0100
committerfinn.williams <finn.williams@arm.com>2019-09-06 13:57:36 +0000
commit9e0deb76fc25be6a0e898f53a115af3aeed9e5b8 (patch)
tree9d7c0157c0c1e6d8c17336c6c8f8c7c64c93aa4d /src/armnn/QuantizerVisitor.hpp
parent366023fcb63d0ade5bd742ede2fc27899296877c (diff)
downloadarmnn-9e0deb76fc25be6a0e898f53a115af3aeed9e5b8.tar.gz
IVGCVSW-3742 Add Quantizer support for ABS
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com> Change-Id: I3f8e716ae432737bb314e26792d18aa518aa1952
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
-rw-r--r--src/armnn/QuantizerVisitor.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp
index 7480a0c0c6..e087fe6a61 100644
--- a/src/armnn/QuantizerVisitor.hpp
+++ b/src/armnn/QuantizerVisitor.hpp
@@ -32,6 +32,8 @@ public:
~QuantizerVisitor() = default;
/// Functions to quantize the individual layers, overridden from ILayerVisitor
+ void VisitAbsLayer(const IConnectableLayer* layer, const char* name = nullptr) override;
+
void VisitActivationLayer(const IConnectableLayer* layer,
const ActivationDescriptor& activationDescriptor,
const char* name = nullptr) override;