Extend prefix as represented by `prefix_states` by one more step. This function is expected to be called during fast decoding of Transformer models. Args: theta: A `.NestedMap` object containing weights' values of this layer and its children layers. encoder_outputs:
(self, theta, encoder_outputs, new_ids, t, prefix_states)
source not stored for this graph (policy: none)