Cast logits to fp32 at the end of TF_T5 (#12332)
Cast logits to fp32 at the end of TF_T5 (#12332)

This change enables tf.keras.mixed_precision with bf16
huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers