Rate this Page
★ ★ ★ ★ ★

torch.nn.attention.list_flash_attention_impls#

torch.nn.attention.list_flash_attention_impls()[source]#

Return the names of all available flash attention implementations.

Return type:

list[str]

On this page
Show Source
PyTorch Libraries