pytorch

38a29324 - [dtensor][2/N] more tensor ops to use strategy propagation (#101203)

[dtensor][2/N] more tensor ops to use strategy propagation (#101203) As titled, this PR adapts a few more tensor ops to use strategy based sharding prop Pull Request resolved: https://github.com/pytorch/pytorch/pull/101203 Approved by: https://github.com/XilunWu