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

Method TrackQActs

lingvo/core/quant_utils.py:193–222  ·  view source on GitHub ↗

Track activations and create variables for quantization as needed. The activations will be associated with a specific named QDomain. Most layers are simple enough to only require a single 'default' QDomain. Additional QDomains can be defined as parameters to control fine grained asp

(self,
                 *act_names: str,
                 shape: Optional[Iterable[int]] = None,
                 feature_axes: Optional[Iterable[int]] = None,
                 domain: str = 'default')

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 2

_GetQDomainMethod · 0.95
TrackQActsMethod · 0.45

Tested by

no test coverage detected