flash_bert

Functions

flash_attention_forward(self, hidden_states, ...)