pytorch/torch/nested/_internal
2024-06-29 00:16:47 +00:00
..
__init__.py
nested_tensor.py Revert "Conversions between strided and jagged layouts for Nested Tensors (#115749)" 2024-06-29 00:16:47 +00:00
ops.py Revert "Conversions between strided and jagged layouts for Nested Tensors (#115749)" 2024-06-29 00:16:47 +00:00
sdpa.py [Nested Tensor]fix sdpa backward for the special case with ragged second batch dim and constant length (#128349) 2024-06-24 22:35:07 +00:00