pytorch

d05ec0ef - [dtensor] add split_with_sizes op (#93957)

[dtensor] add split_with_sizes op (#93957) add the split_with_sizes op, sharing with split op impl Pull Request resolved: https://github.com/pytorch/pytorch/pull/93957 Approved by: https://github.com/XilunWu