Returns logits for sampling ids and the next model states. Args: theta: A `.NestedMap` object containing weights' values of this layer and its children layers. encoder_outputs: a NestedMap computed by encoder. step_ids: A tensor of shape [tgt_batch, 1]. states: A
(self, theta, encoder_outputs, step_ids, states,
num_hyps_per_beam, cur_step)
source not stored for this graph (policy: none)
no test coverage detected