diff options
Diffstat (limited to 'arm_compute/core/PixelValue.h')
| -rw-r--r-- | arm_compute/core/PixelValue.h | 18 |
1 files changed, 8 insertions, 10 deletions
diff --git a/arm_compute/core/PixelValue.h b/arm_compute/core/PixelValue.h index c5f6608163..31bc55098a 100644 --- a/arm_compute/core/PixelValue.h +++ b/arm_compute/core/PixelValue.h @@ -41,11 +41,11 @@ public: } /** Initialize the union with a pixel value of chosen datatype * - * @param[in] v int value. + * @param[in] v value. * @param[in] datatype DataType that @p v have to be stored * @param[in] qinfo (Optional) QuantizationInfo to apply in case of quantized data types to @p v */ - PixelValue(int64_t v, DataType datatype, QuantizationInfo qinfo = QuantizationInfo()) + PixelValue(double v, DataType datatype, QuantizationInfo qinfo = QuantizationInfo()) : PixelValue() { switch(datatype) @@ -57,13 +57,13 @@ public: value.s8 = static_cast<int8_t>(v); break; case DataType::QASYMM8: - value.u8 = quantize_qasymm8(static_cast<uint8_t>(v), qinfo); + value.u8 = quantize_qasymm8(static_cast<float>(v), qinfo); break; case DataType::QASYMM8_SIGNED: - value.s8 = quantize_qasymm8_signed(static_cast<int8_t>(v), qinfo); + value.s8 = quantize_qasymm8_signed(static_cast<float>(v), qinfo); break; case DataType::QSYMM8: - value.s8 = quantize_qsymm8(static_cast<int8_t>(v), qinfo); + value.s8 = quantize_qsymm8(static_cast<float>(v), qinfo); break; case DataType::U16: value.u16 = static_cast<uint16_t>(v); @@ -72,10 +72,10 @@ public: value.s16 = static_cast<int16_t>(v); break; case DataType::QASYMM16: - value.u16 = quantize_qasymm16(static_cast<uint16_t>(v), qinfo); + value.u16 = quantize_qasymm16(static_cast<float>(v), qinfo); break; case DataType::QSYMM16: - value.s16 = quantize_qsymm16(static_cast<int16_t>(v), qinfo); + value.s16 = quantize_qsymm16(static_cast<float>(v), qinfo); break; case DataType::U32: value.u32 = static_cast<uint32_t>(v); @@ -96,10 +96,8 @@ public: value.f32 = static_cast<float>(v); break; case DataType::F64: - value.f64 = static_cast<double>(v); - break; default: - value.s64 = v; + value.f64 = v; break; } } |