From c5e5a04bd26c2d1838828ba291464967a3e20f3f Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Thu, 14 Jan 2021 10:04:56 +0000 Subject: MLCE-325 ArmnnQuantizer incorrectly Quantizes all DataTypes * ArmnnQuantizer incorrectly converts boolean or integer DataTypes to quantized DataTypes. This breaks layers like ArgMinMax where the output contains the index of an element along an axis. Signed-off-by: Mike Kelly Change-Id: I272c3d0f48bf884a2480bfa43eb14ec265fcda6b --- src/armnn/DynamicQuantizationVisitor.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'src/armnn/DynamicQuantizationVisitor.cpp') diff --git a/src/armnn/DynamicQuantizationVisitor.cpp b/src/armnn/DynamicQuantizationVisitor.cpp index 22029102c1..02e7699eed 100644 --- a/src/armnn/DynamicQuantizationVisitor.cpp +++ b/src/armnn/DynamicQuantizationVisitor.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -106,8 +106,7 @@ void DynamicQuantizationVisitor::VisitArgMinMaxLayer(const IConnectableLayer* la { IgnoreUnused(name); IgnoreUnused(desc); - SetRange(layer, 0, -20.f, 20.f); - AddToCalibratedLayers(layer); + AddToNonCalibratedLayers(layer); } void DynamicQuantizationVisitor::VisitBatchNormalizationLayer(const IConnectableLayer* layer, -- cgit v1.2.1