To be safe, always cast to float32 for allreduce. This change may make some workloads run slower.
To be safe, always cast to float32 for allreduce.
This change may make some workloads run slower.

PiperOrigin-RevId: 320481225

tensorflow/mesh,tensorflow/mesh