torch.nn.attention.restore\_flash\_attention\_impl ================================================== .. currentmodule:: torch.nn.attention .. autofunction:: restore_flash_attention_impl