| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_decoder_test.py | 100 beam_width = array_ops.shape(array)[2]
110 array, [max_time, batch_size * beam_width])
112 expected_array, [max_time, batch_size * beam_width])
184 batch_size, beam_width, is_valid=True): argument
194 beam_search_decoder._check_batch_beam(t, batch_size, beam_width)
198 t, batch_size, beam_width)
259 self.beam_width = 3
266 dummy_cell_state = array_ops.zeros([self.batch_size, self.beam_width])
270 array_ops.ones([self.batch_size, self.beam_width])),
272 2, shape=[self.batch_size, self.beam_width], dtype=dtypes.int64),
[all …]
|
| D | beam_search_ops_test.py | 99 beam_width = 15
106 0, high=end_token + 1, size=(max_time, batch_size, beam_width))
108 0, high=beam_width - 1, size=(max_time, batch_size, beam_width))
116 self.assertEqual((max_time, batch_size, beam_width), beams.shape)
123 range(batch_size), range(beam_width)):
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops_gpu.cu.cc | 30 const int32 beam_width, const T* step_ids, in GatherTreeOpKernel() argument
34 CUDA_1D_KERNEL_LOOP(i, batch_size * beam_width) { in GatherTreeOpKernel()
35 const int32 batch = i / beam_width; in GatherTreeOpKernel()
36 const int32 beam = i % beam_width; in GatherTreeOpKernel()
45 (batch_size * beam_width * (time_ix) + beam_width * batch + (beam_ix)) in GatherTreeOpKernel()
53 if (parent < 0 || parent > beam_width) { in GatherTreeOpKernel()
89 const int32 beam_width = parent_ids.dimension(2); in operator ()() local
93 CudaLaunchConfig config = GetCudaLaunchConfig(batch_size * beam_width, d); in operator ()()
96 d.stream(), batch_size, max_time, beam_width, step_ids.data(), in operator ()()
|
| D | beam_search_ops.cc | 114 const int32 beam_width = parent_ids.dimension(2); in operator ()() local
120 const int32 batch = i / beam_width; in operator ()()
121 const int32 beam = i % beam_width; in operator ()()
131 if (parent < 0 || parent > beam_width) { in operator ()()
162 batch_size * beam_width, batch_beam_cost, DoWork); in operator ()()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 145 beam_width = parent_ids.shape.dims[2].value or array_ops.shape(parent_ids)[2]
149 array_ops.expand_dims(math_ops.range(beam_width), 0), 0)
158 end_token=beam_width + 1)
169 math_ops.range(max_time), [-1, 1, 1]), [1, batch_size, beam_width])
171 math_ops.range(batch_size), [-1, 1, 1]), [1, max_time, beam_width])
179 gather_from, [max_time, batch_size, beam_width, -1])
192 def _check_static_batch_beam_maybe(shape, batch_size, beam_width): argument
196 reshaped_shape = tensor_shape.TensorShape([batch_size, beam_width, None])
198 and (shape[0] != batch_size * beam_width
200 and (shape[0] != batch_size or shape[1] != beam_width)))):
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 33 int beam_width; member
44 option->beam_width = m["beam_width"].AsInt32(); in Init()
59 TF_LITE_ENSURE(context, option->beam_width >= top_paths); in Prepare()
173 const int beam_width = option->beam_width; in Eval() local
196 num_classes, beam_width, &beam_scorer, 1 /* batch_size */, in Eval()
|
| D | ctc_beam_search.h | 87 CTCBeamSearchDecoder(int num_classes, int beam_width,
91 beam_width_(beam_width), in CTCDecoder()
92 leaves_(beam_width), in CTCDecoder()
|
| D | ctc_beam_search_decoder_test.cc | 45 int beam_width, int top_paths, in CTCBeamSearchDecoderOpModel() argument
57 fbb.Int("beam_width", beam_width); in CTCBeamSearchDecoderOpModel()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CTCBeamSearchDecoder.pbtxt | 47 name: "beam_width"
55 A scalar >= 0, <= beam_width (controls output size).
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 257 def ctc_beam_search_decoder(inputs, sequence_length, beam_width=100, argument
303 inputs, sequence_length, beam_width=beam_width, top_paths=top_paths,
313 def ctc_beam_search_decoder_v2(inputs, sequence_length, beam_width=100, argument
351 beam_width=beam_width, top_paths=top_paths,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | ctc_decoder_ops_test.py | 235 beam_width=2,
248 beam_width=2,
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 89 CTCBeamSearchDecoder(int num_classes, int beam_width,
93 beam_width_(beam_width), in CTCDecoder()
94 leaves_(beam_width), in CTCDecoder()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 577 op.beam_width = 3; in TEST_F()
584 EXPECT_EQ(op.beam_width, output_toco_op->beam_width); in TEST_F()
|
| D | operator.cc | 1760 fbb->Int("beam_width", op.beam_width); in WriteOptions()
1766 op->beam_width = m["beam_width"].AsInt32(); in ReadOptions()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 121 …argspec: "args=[\'inputs\', \'sequence_length\', \'beam_width\', \'top_paths\', \'merge_repeated\'…
125 …argspec: "args=[\'inputs\', \'sequence_length\', \'beam_width\', \'top_paths\'], varargs=None, key…
|
| D | tensorflow.keras.backend.pbtxt | 121 …argspec: "args=[\'y_pred\', \'input_length\', \'greedy\', \'beam_width\', \'top_paths\'], varargs=…
|
| D | tensorflow.raw_ops.pbtxt | 525 …argspec: "args=[\'inputs\', \'sequence_length\', \'beam_width\', \'top_paths\', \'merge_repeated\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 105 …argspec: "args=[\'inputs\', \'sequence_length\', \'beam_width\', \'top_paths\'], varargs=None, key…
|
| D | tensorflow.keras.backend.pbtxt | 121 …argspec: "args=[\'y_pred\', \'input_length\', \'greedy\', \'beam_width\', \'top_paths\'], varargs=…
|
| D | tensorflow.raw_ops.pbtxt | 525 …argspec: "args=[\'inputs\', \'sequence_length\', \'beam_width\', \'top_paths\', \'merge_repeated\'…
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 1404 beam_width = 2
1411 beam_width=beam_width,
|
| D | backend.py | 5173 def ctc_decode(y_pred, input_length, greedy=True, beam_width=100, top_paths=1): argument
5211 beam_width=beam_width,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 486 int beam_width;
|
| D | import_tensorflow.cc | 2299 op->beam_width = in ConvertCTCBeamSearchDecoderOperator()
|
| D | export_tensorflow.cc | 2034 (*op->mutable_attr())["beam_width"].set_i(src_op.beam_width); in ConvertCTCBeamSearchDecoderOperator()
|