HomeSort by relevance Sort by last modified time
    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