fixing internal_loss for recurrent models with L2 loss.
fixing internal_loss for recurrent models with L2 loss.

PiperOrigin-RevId: 219540943

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