MLCE-161: Added support for NormalizationLayer in ArmnnQuantizer.

* Improved error reporting by using VisitorThrowingPolicy in
   DynamicQuantizationVisitor.

Change-Id: I8a7d0975d448e6c2b6efb5f315ac255eec29f7a2
Signed-off-by: Pablo Tello <pablo.tello@arm.com>

1 files changed, 15 insertions, 3 deletions

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<VisitorNoThrowPolicy>
+class DynamicQuantizationVisitor : public LayerVisitorBase<VisitorThrowingPolicy>
 {
 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