Change default checkpoint saving dtype to float32 instead of bfloat16. Saving
Change default checkpoint saving dtype to float32 instead of bfloat16. Saving

PiperOrigin-RevId: 364201025

tensorflow/mesh,tensorflow/mesh