#include <NetworkQuantizationScheme.hpp>

Inheritance diagram for QAsymmU8QuantizationScheme:

Public Member Functions
OffsetScalePair	ComputeScheme (double min, double max) const override

int	NumBits () const override

DataType	GetDataType () const override

Public Member Functions inherited from IQuantizationScheme
virtual	~IQuantizationScheme ()

Detailed Description

Definition at line 29 of file NetworkQuantizationScheme.hpp.

Member Function Documentation

◆ ComputeScheme()

OffsetScalePair ComputeScheme	(	double	min,
		double	max
	)		const

inlineoverridevirtual

Implements IQuantizationScheme.

Definition at line 31 of file NetworkQuantizationScheme.hpp.

References IQuantizationScheme::NumBits().

Referenced by armnn::QuantizeConstant().

     {
         if (min > max)
         {
             throw InvalidArgumentException("min > max will result in invalid quantization.");
         }
 
         double highest = (1 << NumBits()) - 1;
 
         min = std::min(0.0, min); // min <= 0.0
         max = std::max(0.0, max); // max >= 0.0
 
         // To avoid dividing by zero when quantizing a zero filled tensor
         if (min == 0.0 && max == 0.0)
         {
             max = 1.0;
         }
 
         // Assumes quantization range [0-highest]
         double scale = (max-min) / highest;
         double offset = -min / scale;
 
         // Clamp offset [0-highest]
         offset = std::max(0.0, std::min(highest, offset));
 
         return std::make_pair(static_cast<float>(scale), static_cast<int>(std::round(offset)));
     }