pytorch

789fc4c2 - [dtensor] refactor shape/offset calculation (#95923)

[dtensor] refactor shape/offset calculation (#95923) Shape offset calculation is commonly used and extract them into a separate util Pull Request resolved: https://github.com/pytorch/pytorch/pull/95923 Approved by: https://github.com/fduwjj