pytorch/torch/testing
drisspg 2b43d339fe Make FlexAttention API public (#130755)
# Summary

Makes the prototype API flex_attention public

Pull Request resolved: https://github.com/pytorch/pytorch/pull/130755
Approved by: https://github.com/Chillee
2024-07-16 16:21:25 +00:00
..
_internal Make FlexAttention API public (#130755) 2024-07-16 16:21:25 +00:00
__init__.py
_comparison.py [BE] enable UFMT for torch/storage.py (#127706) 2024-06-27 23:16:24 +00:00
_creation.py
_utils.py