Joel Schlosser 3e2f276a14 Fix to() on non-contiguous NJTs (#137124) ... Called out via torchrec integration: `lengths` is not handled properly. Future work (not related to non-contiguous NJTs): #137275 Pull Request resolved: https://github.com/pytorch/pytorch/pull/137124 Approved by: https://github.com/soulitzer ghstack dependencies: #137030, #137031		2024-10-08 15:11:05 +00:00
..
__init__.py
nested_tensor.py	Fix NJT serialization (#137031)	2024-10-02 21:41:35 +00:00
ops.py	Fix to() on non-contiguous NJTs (#137124)	2024-10-08 15:11:05 +00:00
sdpa.py	[NJT] Support NJT SDPA + meta-device flop counting (#134289)	2024-08-29 03:43:42 +00:00