| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 35 int top_paths; member
46 option->top_paths = m["top_paths"].AsInt32(); in Init()
59 const int top_paths = option->top_paths; in Prepare() local
60 TF_LITE_ENSURE(context, option->beam_width >= top_paths); in Prepare()
63 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 3 * top_paths + 1); in Prepare()
83 for (int i = 0; i < top_paths; ++i) { in Prepare()
89 GetOutputSafe(context, node, i + top_paths, &values)); in Prepare()
92 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i + 2 * top_paths, in Prepare()
99 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, top_paths * 3, in Prepare()
103 log_probability_output_shape_array->data[1] = top_paths; in Prepare()
[all …]
|
| D | ctc_beam_search_decoder_test.cc | 45 int beam_width, int top_paths, in CTCBeamSearchDecoderOpModel() argument
50 for (int i = 0; i < top_paths * 3; ++i) { in CTCBeamSearchDecoderOpModel()
58 fbb.Int("top_paths", top_paths); in CTCBeamSearchDecoderOpModel()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search_test.cc | 108 const int top_paths = 3; in ctc_beam_search_decoding_with_and_without_dictionary() local
114 tensorflow::ctc::CTCBeamSearchDecoder<T> decoder(num_classes, 10 * top_paths, in ctc_beam_search_decoding_with_and_without_dictionary()
120 dictionary_decoder(num_classes, top_paths, &dictionary_scorer); in ctc_beam_search_decoding_with_and_without_dictionary()
168 top_paths); in ctc_beam_search_decoding_with_and_without_dictionary()
172 T score[batch_size][top_paths] = {{0.0}}; in ctc_beam_search_decoding_with_and_without_dictionary()
174 &score[0][0], batch_size, top_paths); in ctc_beam_search_decoding_with_and_without_dictionary()
177 for (int path = 0; path < top_paths; ++path) { in ctc_beam_search_decoding_with_and_without_dictionary()
183 top_paths); in ctc_beam_search_decoding_with_and_without_dictionary()
189 for (int path = 0; path < top_paths; ++path) { in ctc_beam_search_decoding_with_and_without_dictionary()
198 const int top_paths = 3; in ctc_beam_search_decoding_all_beam_elements_have_finite_scores() local
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CTCBeamSearchDecoder.pbtxt | 14 number_attr: "top_paths"
19 number_attr: "top_paths"
24 number_attr: "top_paths"
37 name: "top_paths"
63 number_attr: "top_paths"
68 number_attr: "top_paths"
73 number_attr: "top_paths"
86 name: "top_paths"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CTCBeamSearchDecoder.pbtxt | 14 number_attr: "top_paths"
19 number_attr: "top_paths"
24 number_attr: "top_paths"
37 name: "top_paths"
63 number_attr: "top_paths"
68 number_attr: "top_paths"
73 number_attr: "top_paths"
86 name: "top_paths"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ctc_ops.cc | 154 int32 top_paths; in __anon7dc6794d0402() local
155 TF_RETURN_IF_ERROR(c->GetAttr("top_paths", &top_paths)); in __anon7dc6794d0402()
159 for (int i = 0; i < top_paths; ++i) { // decoded_indices in __anon7dc6794d0402()
162 for (int i = 0; i < top_paths; ++i) { // decoded_values in __anon7dc6794d0402()
166 for (int i = 0; i < top_paths; ++i) { // decoded_shape in __anon7dc6794d0402()
169 c->set_output(out_idx++, c->Matrix(batch_size, top_paths)); in __anon7dc6794d0402()
|
| D | ctc_ops_test.cc | 68 auto set_top_paths = [&op](int top_paths) { in TEST() argument
72 .Attr("top_paths", top_paths) in TEST()
|
| D | ops.pbtxt | 6791 number_attr: "top_paths"
6796 number_attr: "top_paths"
6801 number_attr: "top_paths"
6814 name: "top_paths"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CTCBeamSearchDecoder.pbtxt | 18 A list (length: top_paths) of indices matrices. Matrix j,
26 A list (length: top_paths) of values vectors. Vector j,
34 A list (length: top_paths) of shape vector. Vector j,
42 A matrix, shaped: `(batch_size x top_paths)`. The
53 name: "top_paths"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | ctc_decoder_ops_test.py | 236 top_paths=2)
249 top_paths=3)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 282 int top_paths; in CTCBeamSearchDecoderOp() local
283 OP_REQUIRES_OK(ctx, ctx->GetAttr("top_paths", &top_paths)); in CTCBeamSearchDecoderOp()
284 decode_helper_.SetTopPaths(top_paths); in CTCBeamSearchDecoderOp()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 343 top_paths=1, argument
392 top_paths=top_paths,
406 top_paths=1): argument
446 top_paths=top_paths,
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 644 op.top_paths = 2; in TEST_F()
651 EXPECT_EQ(op.top_paths, output_toco_op->top_paths); in TEST_F()
|
| D | operator.cc | 1415 fbb->Int("top_paths", op.top_paths); in WriteOptions()
1421 op->top_paths = m["top_paths"].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=…
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 1854 top_paths = 2
1861 top_paths=top_paths)
1863 self.assertEqual(len(decode_pred_tf), top_paths)
1865 for i in range(top_paths):
|
| D | backend.py | 6348 def ctc_decode(y_pred, input_length, greedy=True, beam_width=100, top_paths=1): argument
6390 top_paths=top_paths)
|
| /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 | 117 …argspec: "args=[\'y_pred\', \'input_length\', \'greedy\', \'beam_width\', \'top_paths\'], varargs=…
|
| D | tensorflow.raw_ops.pbtxt | 685 …argspec: "args=[\'inputs\', \'sequence_length\', \'beam_width\', \'top_paths\', \'merge_repeated\'…
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 504 int top_paths;
|
| D | import_tensorflow.cc | 2435 op->top_paths = in ConvertCTCBeamSearchDecoderOperator()
2443 for (int i = 0; i < op->top_paths; ++i) { in ConvertCTCBeamSearchDecoderOperator()
|
| D | export_tensorflow.cc | 2069 (*op->mutable_attr())["top_paths"].set_i(src_op.top_paths); in ConvertCTCBeamSearchDecoderOperator()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 7230 number_attr: "top_paths"
7235 number_attr: "top_paths"
7240 number_attr: "top_paths"
7253 name: "top_paths"
|