pytorch
17348b0f - Implement split_with_sizes backward for NT (#110647)

Commit
1 year ago
Implement split_with_sizes backward for NT (#110647) Needed internally. Note that `split_with_sizes()` for NT is currently supported only on `dim=-1`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/110647 Approved by: https://github.com/cpuhrsch, https://github.com/soulitzer ghstack dependencies: #110646
Author
Committer
Parents
Loading