diff options
Diffstat (limited to 'src/armnn/DynamicQuantizationVisitor.cpp')
-rw-r--r-- | src/armnn/DynamicQuantizationVisitor.cpp | 31 |
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, |