xla

c05de860 - Update test_train_spmd_linear_model.py to show clear loss value (#8481)