hub / github.com/tensorflow/lingvo / ToAqtInputs

Method ToAqtInputs

lingvo/core/quant_utils.py:552–588 · view source on GitHub ↗

Quantizes weights and activations for (act * w) matmul AQT style. This only scales, rounds and clips; resulting quantized inputs would be either integer ot integer emulated in float. w_name must have been previously created via TrackQWeight. Args: w_name: Previously created

(self,
                  w_name,
                  act,
                  weight,
                  w_feature_axis,
                  act_distribution=QDistribution.SYMMETRIC,
                  w_expected_scale_shape=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

_ValidateWeightMethod · 0.95

_GetQDomainMethod · 0.95

ToAqtInputsMethod · 0.45

Tested by

no test coverage detected