pytorch/torch/nested
soulitzer 4af4910b1a Reland "Construct NJT without graph breaks" (#133196)
This reverts commit 154d40ca488e6979ce9c2de89d8a35b53129ebea.

and adds changes from https://github.com/pytorch/pytorch/pull/133061

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133196
Approved by: https://github.com/ezyang
ghstack dependencies: #133145
2024-08-14 01:11:13 +00:00
..
_internal Reland "Construct NJT without graph breaks" (#133196) 2024-08-14 01:11:13 +00:00
__init__.py Revert "Conversions between strided and jagged layouts for Nested Tensors (#115749)" 2024-08-12 23:14:19 +00:00