MCPcopy Create free account
hub / github.com/tensorflow/lingvo / ToAqtWeight

Method ToAqtWeight

lingvo/core/quant_utils.py:933–948  ·  view source on GitHub ↗

Quantized weight AQT style. Refer to quantizable_layer.ToAqtWeight. Args: w_name: weight name. w: The weight tensor. feature_axis: axis corresponding to output channel/feature for weights. expected_scale_shape: Optional shape to verify if scale shape is expected.

(self, w_name, w, feature_axis, expected_scale_shape)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

_FlatFPropMethod · 0.45
ToAqtWeightMethod · 0.45
MultiTaskProjectionFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected