HomeSort by relevance Sort by last modified time
    Searched refs:word_indices (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/eager/python/examples/spinn/
data_test.py 233 word_indices, shift_reduce = data.encode_sentence(sentence, word2index)
234 self.assertEqual(np.int64, word_indices.dtype)
235 self.assertEqual((5, 1), word_indices.shape)
data.py 247 word_indices = pad_and_reverse_word_ids(
249 return (word_indices,
  /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
beam_search_decoder.py 579 next_beam_scores, word_indices = nn_ops.top_k(scores_flat, k=next_beam_size)
582 word_indices.set_shape([static_batch_size, beam_width])
586 gather_indices=word_indices,
593 # math_ops.to_int32(word_indices % vocab_size,
598 word_indices, vocab_size, name="next_beam_word_ids")
601 word_indices / vocab_size, name="next_beam_parent_ids")

Completed in 118 milliseconds