hub / github.com/tensorflow/lingvo / BeamSearchDecode

Method BeamSearchDecode

lingvo/core/base_decoder.py:344–357 · view source on GitHub ↗

Performs beam search based decoding. Args: encoder_outputs: the outputs of the encoder. num_hyps_per_beam_override: If set to a value <= 0, this parameter is ignored. If set to a value > 0, then this value will be used to override p.num_hyps_per_beam. Returns:

(self, encoder_outputs, num_hyps_per_beam_override=0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

GetBeamSearchHelperResultsFunction · 0.45

testBeamSearchForceEosInTopKMethod · 0.45

testBeamSearchForceLastChunkEocInTopKMethod · 0.45

testCustomStepIdsMethod · 0.45

BeamSearchDecodeWithThetaMethod · 0.45

StochasticBeamSearchDecodeBiasedMethod · 0.45

_testDecoderBeamSearchDecodeHelperWithOutputMethod · 0.45

_InferenceSubgraph_DefaultMethod · 0.45

testBeamSearchDecodeMethod · 0.45

testBeamSearchDecodeTgtPrefixMethod · 0.45

testBeamSearchDecodeUseZeroAttenStateMethod · 0.45

Calls 1

BeamSearchDecodeWithThetaMethod · 0.95

Tested by 14

GetBeamSearchHelperResultsFunction · 0.36

testBeamSearchForceEosInTopKMethod · 0.36

testBeamSearchForceLastChunkEocInTopKMethod · 0.36

testCustomStepIdsMethod · 0.36

_testDecoderBeamSearchDecodeHelperWithOutputMethod · 0.36

testBeamSearchDecodeMethod · 0.36

testBeamSearchDecodeTgtPrefixMethod · 0.36

testBeamSearchDecodeUseZeroAttenStateMethod · 0.36

testBeamSearchDecodeFeedingAttContextMethod · 0.36

testBeamSearchDisallowMisalignmentMethod · 0.36

testBeamSearchDecodeFactorizedEmbeddingEnabledMethod · 0.36

testSingleTokenFastDecodeMethod · 0.36