Home
last modified time | relevance | path

Searched refs:leaves_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search.h92 leaves_(beam_width), in CTCDecoder()
148 gtl::TopN<BeamEntry*, CTCBeamComparer> leaves_; variable
185 std::unique_ptr<std::vector<BeamEntry*>> branches(leaves_.Extract()); in Decode()
186 leaves_.Reset(); in Decode()
192 leaves_.push(entry); in Decode()
277 std::unique_ptr<std::vector<BeamEntry*>> branches(leaves_.Extract()); in Step()
278 leaves_.Reset(); in Step()
310 leaves_.push(b); in Step()
326 (leaves_.size() < beam_width_ || in Step()
327 prob.total > leaves_.peek_bottom()->newp.total)); in Step()
[all …]
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_beam_search.h94 leaves_(beam_width), in CTCDecoder()
150 gtl::TopN<BeamEntry*, CTCBeamComparer> leaves_; variable
188 std::unique_ptr<std::vector<BeamEntry*>> branches(leaves_.Extract()); in Decode()
189 leaves_.Reset(); in Decode()
195 leaves_.push(entry); in Decode()
280 std::unique_ptr<std::vector<BeamEntry*>> branches(leaves_.Extract()); in Step()
281 leaves_.Reset(); in Step()
313 leaves_.push(b); in Step()
329 (leaves_.size() < beam_width_ || in Step()
330 prob.total > leaves_.peek_bottom()->newp.total)); in Step()
[all …]