| D | beam_search_ops.cc | 121 const int32 beam = i % beam_width; in operator ()() local
127 beams(max_seq_len_b - 1, batch, beam) = in operator ()()
128 step_ids(max_seq_len_b - 1, batch, beam); in operator ()()
129 int32 parent = parent_ids(max_seq_len_b - 1, batch, beam); in operator ()()
135 ", ", level, ", ", beam, ")")); in operator ()()
138 beams(level, batch, beam) = step_ids(level, batch, parent); in operator ()()
147 beams(time, batch, beam) = end_token; in operator ()()
148 } else if (beams(time, batch, beam) == end_token) { in operator ()()
|