def next_inputs(self, time, sample_ids): finished = math_ops.logical_or( tf.greater_equal(time + 1, self.max_step), tf.equal(self.eos_id, sample_ids)) return finished, self.lookup(sample_ids)