diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/TypesUtils.cpp | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/src/armnn/TypesUtils.cpp b/src/armnn/TypesUtils.cpp index 74ac231bc9..d419ef84c7 100644 --- a/src/armnn/TypesUtils.cpp +++ b/src/armnn/TypesUtils.cpp @@ -32,8 +32,14 @@ QuantizedType armnn::Quantize(float value, float scale, int32_t offset) static_assert(IsQuantizedType<QuantizedType>(), "Not an integer type."); constexpr QuantizedType max = std::numeric_limits<QuantizedType>::max(); constexpr QuantizedType min = std::numeric_limits<QuantizedType>::lowest(); - ARMNN_ASSERT(scale != 0.f); - ARMNN_ASSERT(!std::isnan(value)); + if (scale == 0.f) + { + throw armnn::InvalidArgumentException("Quantize: Scale cannot be 0.f"); + } + if (std::isnan(value)) + { + throw armnn::InvalidArgumentException("Quantize: Value is NaN"); + } float clampedValue = std::min(std::max((static_cast<float>(offset) + static_cast<float>(round(value/scale))), static_cast<float>(min)), static_cast<float>(max)); @@ -46,8 +52,14 @@ template <typename QuantizedType> float armnn::Dequantize(QuantizedType value, float scale, int32_t offset) { static_assert(IsQuantizedType<QuantizedType>(), "Not an integer type."); - ARMNN_ASSERT(scale != 0.f); - ARMNN_ASSERT(!IsNan(value)); + if (scale == 0.f) + { + throw armnn::InvalidArgumentException("Dequantize: Scale cannot be 0.f"); + } + if (std::isnan(value)) + { + throw armnn::InvalidArgumentException("Dequantize: Value is NaN"); + } return (armnn::numeric_cast<float>(value - offset)) * scale; } |