Use non-causal attention for non-autoregressive decoding from latents
Use non-causal attention for non-autoregressive decoding from latents

PiperOrigin-RevId: 192137977

tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor,tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor,vthorsteinsson/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,vthorsteinsson/tensor2tensor