My Project
|
#include <UniformSupport.h>
Public Member Functions | |
UniformQuantizedPerAxisValueConverter (UniformQuantizedPerAxisType uniformType) | |
ElementsAttr | convert (Attribute realValue) |
An utility class to quantize an attribute by the per-axis quantization parameters. The size of the quantization dim in the converted elements attribute should matche the size of of scales/zeroPoints vectors in the quantization parameters.
|
inlineexplicit |
ElementsAttr UniformQuantizedPerAxisValueConverter::convert | ( | Attribute | realValue | ) |
Quantize an Attribute by the quantization parameters. Return nullptr if the conversion fails or the input array isn't an ElementsAttr.