OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ecc5e05854
pytorch/torch/nested/_internal
History
Joel Schlosser ecc5e05854 Refactor NJT min / max seqlen handling for convenience (#138130) 
There's an annoying pattern emerging for pulling out the NJT min / max seqlen ints if they exist without computing / caching if they don't. This PR introduces private convenience functions to simplify handling this and avoiding redundant checks.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/138130
Approved by: https://github.com/soulitzer
2024-10-17 17:28:39 +00:00
..
__init__.py
nested_tensor.py Refactor NJT min / max seqlen handling for convenience (#138130) 2024-10-17 17:28:39 +00:00
ops.py Refactor NJT min / max seqlen handling for convenience (#138130) 2024-10-17 17:28:39 +00:00
sdpa.py Refactor NJT min / max seqlen handling for convenience (#138130) 2024-10-17 17:28:39 +00:00