Searched refs:p_num (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 126 const int32_t p_num = num_entries[p]; in StoreAllDecodedSequences() local
130 TF_LITE_ENSURE_OK(context, Resize(context, {p_num, 2}, indices)); in StoreAllDecodedSequences()
133 TF_LITE_ENSURE_OK(context, Resize(context, {p_num}, values)); in StoreAllDecodedSequences()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 134 const int64 p_num = num_entries[p]; in StoreAllDecodedSequences() local
137 decoded_indices->allocate(p, TensorShape({p_num, 2}), &p_indices); in StoreAllDecodedSequences()
139 s = decoded_values->allocate(p, TensorShape({p_num}), &p_values); in StoreAllDecodedSequences()
|