xla

f5c00624 - Make clip_grad_norm_ follow the input's dtype (#7205)

Make clip_grad_norm_ follow the input's dtype (#7205) Summary: The correct clipped value uses fp32 regardless of the input dtype. Let's make it follow the input dtype. Test Plan: python test/test_operations.py -v -k test_clip_grad_norm_