OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:beam_state
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
beam_search_decoder_test.py
119
beam_state
= beam_search_decoder.BeamSearchDecoderState(
145
beam_state
=
beam_state
,
153
[outputs, next_beam_state,
beam_state
, log_probs])
174
beam_state
= beam_search_decoder.BeamSearchDecoderState(
200
beam_state
=
beam_state
,
208
[outputs, next_beam_state,
beam_state
, log_probs])
275
beam_state
= beam_search_decoder.BeamSearchDecoderState(
298
beam_state
=beam_state
[
all
...]
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
beam_search_decoder.py
503
beam_state
=state,
518
def _beam_search_step(time, logits, next_cell_state,
beam_state
, batch_size,
530
beam_state
: Current state of the beam search.
543
prediction_lengths =
beam_state
.lengths
544
previously_finished =
beam_state
.finished
550
total_probs = array_ops.expand_dims(
beam_state
.log_probs, 2) + step_log_probs
623
gather_from=
beam_state
.lengths,
Completed in 78 milliseconds