36 template <
typename Tin,
typename Tout>
50 switch(output_data_type)
54 #pragma omp parallel for 56 for(
int i = 0; i <
src.num_elements(); ++i)
63 #pragma omp parallel for 65 for(
int i = 0; i <
src.num_elements(); ++i)
76 #pragma omp parallel for 78 for(
int i = 0; i <
src.num_elements(); ++i)
93 return quantization_layer<float, uint8_t>(src_tmp, output_data_type, quantization_info);
100 return quantization_layer<float, int8_t>(src_tmp, output_data_type, quantization_info);
107 return quantization_layer<float, uint8_t>(src_tmp, output_data_type, quantization_info);
114 return quantization_layer<float, int8_t>(src_tmp, output_data_type, quantization_info);
121 return quantization_layer<float, uint16_t>(src_tmp, output_data_type, quantization_info);
SimpleTensor< Tout > quantization_layer(const SimpleTensor< Tin > &src, DataType output_data_type, const QuantizationInfo &quantization_info)
uint8_t quantize_qasymm8(float value, const INFO_TYPE &qinfo, RoundingPolicy rounding_policy=RoundingPolicy::TO_NEAREST_UP)
Quantize a value given an unsigned 8-bit asymmetric quantization scheme.
#define ARM_COMPUTE_ERROR(msg)
Print the given message then throw an std::runtime_error.
quantized, asymmetric fixed-point 16-bit number
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
Quantization information.
int8_t quantize_qasymm8_signed(float value, const INFO_TYPE &qinfo, RoundingPolicy rounding_policy=RoundingPolicy::TO_NEAREST_UP)
Quantize a value given a signed 8-bit asymmetric quantization scheme.
quantized, asymmetric fixed-point 8-bit number unsigned
UniformQuantizationInfo uniform() const
Return per layer quantization info.
Simple tensor object that stores elements in a consecutive chunk of memory.
Rounds to nearest value; half rounds to nearest even.
const QuantizationInfo qinfo
quantized, asymmetric fixed-point 8-bit number signed
DataType
Available data types.
Truncates the least significant values that are lost in operations.
uint16_t quantize_qasymm16(float value, const UniformQuantizationInfo &qinfo, RoundingPolicy rounding_policy=RoundingPolicy::TO_NEAREST_UP)
Quantize a value given a 16-bit asymmetric quantization scheme.