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

Method TrackQWeight

lingvo/core/quant_utils.py:224–268  ·  view source on GitHub ↗

Creates Quantized weights for later use. Weight that will later be quantized must be created first, preferably in _CreateLayerVariables(). Args: weight_name: Positional parameters are taken to be weight names to create. shape: Shape of the weight. feature_axis: axis c

(self,
                   weight_name,
                   shape,
                   feature_axis,
                   domain: str = 'default',
                   *,
                   tensor_split_dims_mapping=None,
                   device_mesh=None,
                   legacy_aqt_weight_name=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 3

_GetQDomainMethod · 0.95
_CreateChildContextMethod · 0.80
TrackQWeightMethod · 0.45

Tested by

no test coverage detected