Searched defs:beam_width (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 192 def _check_static_batch_beam_maybe(shape, batch_size, beam_width): argument
211 def _check_batch_beam(t, batch_size, beam_width): argument
245 beam_width, argument
637 beam_width, argument
797 beam_width, argument
943 beam_width, end_token, length_penalty_weight, argument
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops_gpu.cu.cc | 30 const int32 beam_width, const T* step_ids, in GatherTreeOpKernel()
89 const int32 beam_width = parent_ids.dimension(2); in operator ()() local
|
| D | beam_search_ops.cc | 114 const int32 beam_width = parent_ids.dimension(2); in operator ()() local
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 33 int beam_width; member
173 const int beam_width = option->beam_width; in Eval() local
|
| D | ctc_beam_search_decoder_test.cc | 45 int beam_width, int top_paths, in CTCBeamSearchDecoderOpModel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 257 def ctc_beam_search_decoder(inputs, sequence_length, beam_width=100, argument
313 def ctc_beam_search_decoder_v2(inputs, sequence_length, beam_width=100, argument
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 184 batch_size, beam_width, is_valid=True): argument
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5173 def ctc_decode(y_pred, input_length, greedy=True, beam_width=100, top_paths=1): argument
|