aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Tensor.cpp
diff options
context:
space:
mode:
authorColm Donelan <colm.donelan@arm.com>2024-03-29 16:00:24 +0000
committerColm Donelan <colm.donelan@arm.com>2024-04-17 13:46:29 +0000
commit4f1771ab4d321afba9f5a52411855b5dc33bf247 (patch)
treea5772debacfaaea83b078d923633c10715284d98 /src/armnn/Tensor.cpp
parent4e74df277b80db0be5c19b07d9f59575dc7ef3d6 (diff)
downloadarmnn-4f1771ab4d321afba9f5a52411855b5dc33bf247.tar.gz
MLCE-1248 Removing limitations on zero scale value in quantization.
Currently Arm NN will fail to load models containing quantization scale value of zero. Signed-off-by: Colm Donelan <colm.donelan@arm.com> Change-Id: Ifefcee1279b8667da63d1aa7d42e5d44875f9fbe
Diffstat (limited to 'src/armnn/Tensor.cpp')
-rw-r--r--src/armnn/Tensor.cpp9
1 files changed, 8 insertions, 1 deletions
diff --git a/src/armnn/Tensor.cpp b/src/armnn/Tensor.cpp
index 3b116d90e0..f75fc60ef7 100644
--- a/src/armnn/Tensor.cpp
+++ b/src/armnn/Tensor.cpp
@@ -465,8 +465,15 @@ float TensorInfo::GetQuantizationScale() const
// NOTE: old default for backward compatibility
return 1.0f;
}
-
+ // If this tensor includes multiples scales then you should be calling GetQuantizationScales.
+ // This should be an exception not an assert but unfortunately it breaks many tests.
+ // ToDo: IVGCVSW-8323
ARMNN_ASSERT(!HasMultipleQuantizationScales());
+// if (HasMultipleQuantizationScales())
+// {
+// throw RuntimeException("Invalid call to GetQuantizationScale on a tensor with multiple scale values. Use "
+// "GetQuantizationScales instead.");
+// }
return m_Quantization.m_Scales[0];
}