pytorch

4f9d4e1a - [DTensor][XLA] refactor DTensor _xla API (#113214)

[DTensor][XLA] refactor DTensor _xla API (#113214) In response to the change pytorch/xla#5776 and #92909 Pull Request resolved: https://github.com/pytorch/pytorch/pull/113214 Approved by: https://github.com/wanchaol