Searched refs:beam_scorer_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search.h | 93 beam_scorer_(scorer) { in CTCDecoder()
115 BaseBeamScorer<CTCBeamState>* GetBeamScorer() const { return beam_scorer_; } in GetBeamScorer()
150 BaseBeamScorer<CTCBeamState>* beam_scorer_; variable
189 beam_scorer_->ExpandStateEnd(&entry->state); in Decode()
191 beam_scorer_->GetStateEndExpansionScore(entry->state); in Decode()
298 beam_scorer_->GetStateExpansionScore(b->state, previous)); in Step()
352 beam_scorer_->ExpandState(b->state, b->label, &c.state, c.label); in Step()
355 beam_scorer_->GetStateExpansionScore(c.state, previous); in Step()
393 beam_scorer_->InitializeState(&beam_root_->RootEntry()->state); in Reset()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 95 beam_scorer_(CHECK_NOTNULL(scorer)) {
117 return beam_scorer_; in GetBeamScorer()
153 BaseBeamScorer<T, CTCBeamState>* beam_scorer_; variable
194 beam_scorer_->ExpandStateEnd(&entry->state); in Decode()
196 beam_scorer_->GetStateEndExpansionScore(entry->state); in Decode()
303 beam_scorer_->GetStateExpansionScore(b->state, previous)); in Step()
357 beam_scorer_->ExpandState(b->state, b->label, &c.state, c.label); in Step()
360 beam_scorer_->GetStateExpansionScore(c.state, previous); in Step()
398 beam_scorer_->InitializeState(&beam_root_->RootEntry()->state); in Reset()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 323 &beam_scorer_, 1 /* batch_size */, in Compute()
359 typename ctc::CTCBeamSearchDecoder<T>::DefaultBeamScorer beam_scorer_; member in tensorflow::CTCBeamSearchDecoderOp
|