Skip to content
Lightning IR
Toggle navigation menu
⌘
K
Lightning IR
/
API Reference
/
lightning_ir
/
flash
/
flash_bert
flash_bert
Functions
flash_attention_forward
(self, hidden_states, ...)