xla

172097b6 - Early exit for clip_grad_norm_ (#7621)

Early exit for clip_grad_norm_ (#7621) Summary: Adds an early exit for clip_grad_norm_. Test Plan: PJRT_DEVICE=TPU python test/test_operations.py -v -k test_clip_grad_norm_