From 5c391dcdc3dd2c4e0f246be3eeb7a089d3c9323d Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Thu, 12 Mar 2020 10:58:39 +0000 Subject: MLCE-161: Added support for NormalizationLayer in ArmnnQuantizer. * Improved error reporting by using VisitorThrowingPolicy in DynamicQuantizationVisitor. Change-Id: I8a7d0975d448e6c2b6efb5f315ac255eec29f7a2 Signed-off-by: Pablo Tello --- src/armnn/DynamicQuantizationVisitor.hpp | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) (limited to 'src/armnn/DynamicQuantizationVisitor.hpp') diff --git a/src/armnn/DynamicQuantizationVisitor.hpp b/src/armnn/DynamicQuantizationVisitor.hpp index 43768fd077..358e47187e 100644 --- a/src/armnn/DynamicQuantizationVisitor.hpp +++ b/src/armnn/DynamicQuantizationVisitor.hpp @@ -16,14 +16,26 @@ namespace armnn { /// Visitor class to establish min/max ranges based on the type of the layer -class DynamicQuantizationVisitor : public LayerVisitorBase +class DynamicQuantizationVisitor : public LayerVisitorBase { public: DynamicQuantizationVisitor(RangeTracker& rangeTracker, Graph& graph); ~DynamicQuantizationVisitor() = default; /// Functions to set the Range on a per-layer-type basis - void VisitAdditionLayer(const IConnectableLayer* layer, const char* name = nullptr) override; + void VisitAbsLayer(const IConnectableLayer* layer, + const char* name = nullptr) override; + + void VisitAdditionLayer(const IConnectableLayer* layer, + const char* name = nullptr) override; + + void VisitArgMinMaxLayer(const IConnectableLayer* layer, + const ArgMinMaxDescriptor& desc, + const char* name = nullptr) override; + + void VisitNormalizationLayer(const IConnectableLayer* layer, + const NormalizationDescriptor& desc, + const char* name = nullptr) override ; void VisitBatchNormalizationLayer(const IConnectableLayer* layer, const BatchNormalizationDescriptor& desc, @@ -134,4 +146,4 @@ private: void RemoveDebugLayers(); }; -} //namespace armnn \ No newline at end of file +} //namespace armnn -- cgit v1.2.1