xla

fcef5273 - [Functionalization] Lower diagonal_scatter properly (#4784)

[Functionalization] Lower diagonal_scatter properly (#4784) Summary: This change replaces the old implementation with DiagonalViewUpdate. Test Plan: PJRT_DEVICE=CPU python ../test/test_view_ops.py -v -k TestViewOpsXLA.test_diagonal