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

Method ExtendStep

lingvo/tasks/mt/decoder.py:2622–2706  ·  view source on GitHub ↗

Extend prefix as represented by `prefix_states` by one more step. This function is expected to be called during fast decoding of Transformer models. Args: theta: A `.NestedMap` object containing weights' values of this layer and its children layers. encoder_outputs:

(self,
                 theta,
                 encoder_outputs,
                 new_ids,
                 time_step,
                 prefix_states,
                 use_short_seq_opt=False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

Calls 6

DeepCopyMethod · 0.80
EmbLookupMethod · 0.45
FPropWithPositionMethod · 0.45
FPropMethod · 0.45
ExtendStepMethod · 0.45

Tested by

no test coverage detected