xla

Cherry-pick: Enable xla:gpu autocast for bfloat16 if not restricted (#5570)

#5591

Merged