Searched refs:p_num (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 136 const int64 p_num = num_entries[p]; in StoreAllDecodedSequences() local
139 decoded_indices->allocate(p, TensorShape({p_num, 2}), &p_indices); in StoreAllDecodedSequences()
141 s = decoded_values->allocate(p, TensorShape({p_num}), &p_values); in StoreAllDecodedSequences()
159 << "values_t should not be nullptr: p_num=" << p_num in StoreAllDecodedSequences()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search_decoder.cc | 136 const int32_t p_num = num_entries[p]; in StoreAllDecodedSequences() local
141 TF_LITE_ENSURE_OK(context, Resize(context, {p_num, 2}, indices)); in StoreAllDecodedSequences()
146 TF_LITE_ENSURE_OK(context, Resize(context, {p_num}, values)); in StoreAllDecodedSequences()
|