[XLA] [NFC] Don't use deprecated tsl Status helpers
[XLA] [NFC] Don't use deprecated tsl Status helpers

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10724 from Tixxx:tixxx/shape_crash 89981b6576a22134831ddfc528fad900129d5058
