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)) { in CTCDecoder()
117 BaseBeamScorer<CTCBeamState>* GetBeamScorer() const { return beam_scorer_; } in GetBeamScorer()
152 BaseBeamScorer<CTCBeamState>* beam_scorer_; variable
192 beam_scorer_->ExpandStateEnd(&entry->state); in Decode()
194 beam_scorer_->GetStateEndExpansionScore(entry->state); in Decode()
301 beam_scorer_->GetStateExpansionScore(b->state, previous)); in Step()
355 beam_scorer_->ExpandState(b->state, b->label, &c.state, c.label); in Step()
358 beam_scorer_->GetStateExpansionScore(c.state, previous); in Step()
396 beam_scorer_->InitializeState(&beam_root_->RootEntry()->state); in Reset()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 302 &beam_scorer_, 1 /* batch_size */, in Compute()
338 ctc::CTCBeamSearchDecoder<>::DefaultBeamScorer beam_scorer_; member in tensorflow::CTCBeamSearchDecoderOp
|