aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/DynamicQuantizationVisitor.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/DynamicQuantizationVisitor.cpp')
-rw-r--r--src/armnn/DynamicQuantizationVisitor.cpp31
1 files changed, 30 insertions, 1 deletions
diff --git a/src/armnn/DynamicQuantizationVisitor.cpp b/src/armnn/DynamicQuantizationVisitor.cpp
index 862a926abc..0b03a3861d 100644
--- a/src/armnn/DynamicQuantizationVisitor.cpp
+++ b/src/armnn/DynamicQuantizationVisitor.cpp
@@ -83,9 +83,28 @@ void DynamicQuantizationVisitor::VisitNonCalibratedLayers() {
}
}
-void DynamicQuantizationVisitor::VisitAdditionLayer(const IConnectableLayer* layer, const char* name)
+void DynamicQuantizationVisitor::VisitAdditionLayer(const IConnectableLayer* layer,
+ const char* name)
+{
+ IgnoreUnused(name);
+ SetRange(layer, 0, -20.f, 20.f);
+ AddToCalibratedLayers(layer);
+}
+
+void DynamicQuantizationVisitor::VisitAbsLayer(const IConnectableLayer* layer,
+ const char* name)
+{
+ IgnoreUnused(name);
+ SetRange(layer, 0, -20.f, 20.f);
+ AddToCalibratedLayers(layer);
+}
+
+void DynamicQuantizationVisitor::VisitArgMinMaxLayer(const IConnectableLayer* layer,
+ const ArgMinMaxDescriptor& desc,
+ const char* name)
{
IgnoreUnused(name);
+ IgnoreUnused(desc);
SetRange(layer, 0, -20.f, 20.f);
AddToCalibratedLayers(layer);
}
@@ -108,6 +127,16 @@ void DynamicQuantizationVisitor::VisitBatchNormalizationLayer(const IConnectable
AddToCalibratedLayers(layer);
}
+void DynamicQuantizationVisitor::VisitNormalizationLayer(const IConnectableLayer* layer,
+ const NormalizationDescriptor& desc,
+ const char* name)
+{
+ IgnoreUnused(desc);
+ IgnoreUnused(name);
+ SetRange(layer, 0, -15.0f, 15.0f);
+ AddToCalibratedLayers(layer);
+}
+
void DynamicQuantizationVisitor::VisitConvolution2dLayer(const IConnectableLayer* layer,
const Convolution2dDescriptor& convolution2dDescriptor,
const ConstTensor& weights,