diff options
author | Pablo Tello <pablo.tello@arm.com> | 2020-03-12 10:58:39 +0000 |
---|---|---|
committer | Pablo Tello <pablo.tello@arm.com> | 2020-03-16 14:22:58 +0000 |
commit | 5c391dcdc3dd2c4e0f246be3eeb7a089d3c9323d (patch) | |
tree | 392fbbc9849fdd34353a295cb516a72cb0663374 /src/armnn/DynamicQuantizationVisitor.hpp | |
parent | fdad393d5611dad69f65f9765c9472fc41cbcfe4 (diff) | |
download | armnn-5c391dcdc3dd2c4e0f246be3eeb7a089d3c9323d.tar.gz |
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>
Diffstat (limited to 'src/armnn/DynamicQuantizationVisitor.hpp')
-rw-r--r-- | src/armnn/DynamicQuantizationVisitor.hpp | 18 |
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 |