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

Method _InferenceSubgraph_Default

lingvo/tasks/lm/model.py:197–276  ·  view source on GitHub ↗

Default inference subgraph. Returns: (fetches, feeds): - fetches: A dictionary of fetches, containing: - log_pplx_per_token: A matrix of shape [batch, time]. [i, j] is i-th input text's j-th token's log prob. - paddings: A matrix of shape [batch, time]. T

(self)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

InferenceMethod · 0.95

Calls 5

StringsToIdsMethod · 0.45
IdsToStringsMethod · 0.45
FPropDefaultThetaMethod · 0.45
zero_stateMethod · 0.45

Tested by

no test coverage detected