| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| beam_search_ops_gpu.cu.cc | 30 const int32 beam_width, const T* step_ids,
34 CUDA_1D_KERNEL_LOOP(i, batch_size * beam_width) {
35 const int32 batch = i / beam_width;
36 const int32 beam = i % beam_width;
45 (batch_size * beam_width * (time_ix) + beam_width * batch + (beam_ix))
53 if (parent < 0 || parent > beam_width) {
89 const int32 beam_width = parent_ids.dimension(2); local
93 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * beam_width, d);
97 batch_size, max_time, beam_width,
[all...] |
| beam_search_ops.cc | 114 const int32 beam_width = parent_ids.dimension(2); local
120 const int32 batch = i / beam_width;
121 const int32 beam = i % beam_width;
131 if (parent < 0 || parent > beam_width) {
162 batch_size * beam_width, batch_beam_cost, DoWork);
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| beam_search_decoder_test.py | 45 # (max_time = 3, batch_size = 2, beam_width = 3)
47 # create (batch_size, max_time, beam_width) matrix and transpose it
112 self.beam_width = 3
118 dummy_cell_state = array_ops.zeros([self.batch_size, self.beam_width])
122 array_ops.ones([self.batch_size, self.beam_width])),
124 2, shape=[self.batch_size, self.beam_width], dtype=dtypes.int64),
126 [self.batch_size, self.beam_width], dtype=dtypes.bool))
128 logits_ = np.full([self.batch_size, self.beam_width, self.vocab_size],
147 beam_width=self.beam_width,
[all...] |
| beam_search_ops_test.py | 101 beam_width = 15
108 0, high=end_token + 1, size=(max_time, batch_size, beam_width))
110 0, high=beam_width - 1, size=(max_time, batch_size, beam_width))
118 self.assertEqual((max_time, batch_size, beam_width), beams.shape)
125 range(batch_size), range(beam_width)):
|
| /external/tensorflow/tensorflow/python/ops/ |
| ctc_ops.py | 235 def ctc_beam_search_decoder(inputs, sequence_length, beam_width=100,
240 `ctc_beam_search_decoder` with `top_paths=1` and `beam_width=1` (but
256 beam_width: An int scalar >= 0 (beam search beam width).
257 top_paths: An int scalar >= 0, <= beam_width (controls output size).
276 inputs, sequence_length, beam_width=beam_width, top_paths=top_paths,
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| beam_search_decoder.py | 68 `[batch_size, T, beam_width]` (or `[T, batch_size, beam_width]` if
139 - The encoder output has been tiled to `beam_width` via
142 wrapper is equal to `true_batch_size * beam_width`.
151 encoder_outputs, multiplier=beam_width)
153 encoder_final_state, multiplier=beam_width)
155 sequence_length, multiplier=beam_width)
162 dtype, batch_size=true_batch_size * beam_width)
174 beam_width,
186 beam_width: Python integer, the number of beams
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| ctc_decoder_ops_test.py | 232 beam_width=2,
245 beam_width=2,
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| ctc_beam_search.h | 82 // The beam search decoder is constructed specifying the beam_width (number of
88 CTCBeamSearchDecoder(int num_classes, int beam_width,
92 beam_width_(beam_width),
93 leaves_(beam_width),
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| backend_test.py | [all...] |
| backend.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
