Searched refs:input_list_t (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | ctc_decoder_ops.cc | 202 std::vector<typename TTypes<T>::UnalignedConstMatrix> input_list_t; in Compute() local 213 input_list_t.reserve(max_time); in Compute() 215 input_list_t.emplace_back(inputs_t.data() + t * batch_size * num_classes, in Compute() 236 -RowMax<T>(input_list_t[t], b, &max_class_indices); in Compute() 314 std::vector<typename TTypes<T>::UnalignedConstMatrix> input_list_t; in Compute() local 316 input_list_t.reserve(max_time); in Compute() 318 input_list_t.emplace_back(inputs_t.data() + t * batch_size * num_classes, in Compute() 336 input_chip_t = input_list_t[t].chip(b, 0); in Compute()
|
D | ctc_loss_op.cc | 186 std::vector<InputMap> input_list_t; in Compute() local 189 input_list_t.emplace_back(inputs_t.data() + t * batch_size * num_classes, in Compute() 203 seq_len_t, labels_t, input_list_t, in Compute()
|
/external/tensorflow/tensorflow/lite/experimental/kernels/ |
D | ctc_beam_search_decoder.cc | 203 std::vector<optimized_ops::TTypes<float>::UnalignedConstMatrix> input_list_t; in Eval() local 206 input_list_t.emplace_back( in Eval() 237 input_chip_t = input_list_t[t].chip(b, 0); in Eval()
|