| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 34 int top_paths; member
45 option->top_paths = m["top_paths"].AsInt32(); in Init()
58 const int top_paths = option->top_paths; in Prepare() local
59 TF_LITE_ENSURE(context, option->beam_width >= top_paths); in Prepare()
62 TF_LITE_ENSURE_EQ(context, NumOutputs(node), 3 * top_paths + 1); in Prepare()
79 for (int i = 0; i < top_paths; ++i) { in Prepare()
82 TfLiteTensor* values = GetOutput(context, node, i + top_paths); in Prepare()
84 TfLiteTensor* output_shape = GetOutput(context, node, i + 2 * top_paths); in Prepare()
90 GetOutput(context, node, top_paths * 3); in Prepare()
93 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 | 106 const int top_paths = 3; in TEST() local
111 CTCBeamSearchDecoder<> decoder(num_classes, 10 * top_paths, &default_scorer); in TEST()
116 num_classes, top_paths, &dictionary_scorer); in TEST()
159 std::vector<CTCDecoder::Output> outputs(top_paths); in TEST()
163 float score[batch_size][top_paths] = {{0.0}}; in TEST()
164 Eigen::Map<Eigen::MatrixXf> scores(&score[0][0], batch_size, top_paths); in TEST()
167 for (int path = 0; path < top_paths; ++path) { in TEST()
172 std::vector<CTCDecoder::Output> dict_outputs(top_paths); in TEST()
178 for (int path = 0; path < top_paths; ++path) { in TEST()
186 const int top_paths = 3; in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ctc_ops.cc | 114 int32 top_paths; in __anon78c05cd60302() local
115 TF_RETURN_IF_ERROR(c->GetAttr("top_paths", &top_paths)); in __anon78c05cd60302()
119 for (int i = 0; i < top_paths; ++i) { // decoded_indices in __anon78c05cd60302()
122 for (int i = 0; i < top_paths; ++i) { // decoded_values in __anon78c05cd60302()
126 for (int i = 0; i < top_paths; ++i) { // decoded_shape in __anon78c05cd60302()
129 c->set_output(out_idx++, c->Matrix(batch_size, top_paths)); in __anon78c05cd60302()
|
| 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 | 5052 number_attr: "top_paths"
5057 number_attr: "top_paths"
5062 number_attr: "top_paths"
5075 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/ops/ |
| D | ctc_ops.py | 258 top_paths=1, merge_repeated=True): argument
303 inputs, sequence_length, beam_width=beam_width, top_paths=top_paths,
314 top_paths=1): argument
351 beam_width=beam_width, top_paths=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 | 262 int top_paths; in CTCBeamSearchDecoderOp() local
263 OP_REQUIRES_OK(ctx, ctx->GetAttr("top_paths", &top_paths)); in CTCBeamSearchDecoderOp()
264 decode_helper_.SetTopPaths(top_paths); in CTCBeamSearchDecoderOp()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator_test.cc | 578 op.top_paths = 2; in TEST_F()
585 EXPECT_EQ(op.top_paths, output_toco_op->top_paths); in TEST_F()
|
| D | operator.cc | 1761 fbb->Int("top_paths", op.top_paths); in WriteOptions()
1767 op->top_paths = m["top_paths"].AsInt32(); in ReadOptions()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 1405 top_paths = 2
1412 top_paths=top_paths)
1414 self.assertEqual(len(decode_pred_tf), top_paths)
1416 for i in range(top_paths):
|
| D | backend.py | 5173 def ctc_decode(y_pred, input_length, greedy=True, beam_width=100, top_paths=1): argument
5212 top_paths=top_paths)
|
| /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/lite/toco/ |
| D | model.h | 487 int top_paths;
|
| D | import_tensorflow.cc | 2301 op->top_paths = in ConvertCTCBeamSearchDecoderOperator()
2309 for (int i = 0; i < op->top_paths; ++i) { in ConvertCTCBeamSearchDecoderOperator()
|
| D | export_tensorflow.cc | 2035 (*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"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4372 … *Scope, inputs tf.Output, sequence_length tf.Output, beam_width int64, top_paths int64, optional …
4376 attrs := map[string]interface{}{"beam_width": beam_width, "top_paths": top_paths}
|