diff options
Diffstat (limited to 'tests/validation/reference/DequantizationLayer.cpp')
-rw-r--r-- | tests/validation/reference/DequantizationLayer.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/validation/reference/DequantizationLayer.cpp b/tests/validation/reference/DequantizationLayer.cpp index df50c14ec7..286a609d79 100644 --- a/tests/validation/reference/DequantizationLayer.cpp +++ b/tests/validation/reference/DequantizationLayer.cpp @@ -34,14 +34,14 @@ namespace reference template <typename T> SimpleTensor<T> dequantization_layer(const SimpleTensor<uint8_t> &src) { - const DataType dst_data_type = std::is_same<T, float>::value ? DataType::F32 : DataType::F16; - const QuantizationInfo &quantization_info = src.quantization_info(); + const DataType dst_data_type = std::is_same<T, float>::value ? DataType::F32 : DataType::F16; + const UniformQuantizationInfo &quantization_info = src.quantization_info().uniform(); SimpleTensor<T> dst{ src.shape(), dst_data_type }; for(int i = 0; i < src.num_elements(); ++i) { - dst[i] = static_cast<T>(quantization_info.dequantize(src[i])); + dst[i] = static_cast<T>(dequantize_qasymm8(src[i], quantization_info)); } return dst; |