pytorch/torch/nested
Joel Schlosser bf62511e07 Reshape decomposition for jagged layout NT (#115191)
No more segfault from using `reshape()` on jagged NT :)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115191
Approved by: https://github.com/cpuhrsch, https://github.com/soulitzer
2023-12-18 22:34:41 +00:00
..
_internal Reshape decomposition for jagged layout NT (#115191) 2023-12-18 22:34:41 +00:00
__init__.py Implement narrow from a regular tensor to jagged tensor (#112770) 2023-11-13 19:09:59 +00:00