Searched defs:can_use_flash_attention (Results 1 – 2 of 2) sorted by relevance
357 def can_use_flash_attention(params: SDPAParams, debug: bool = False) -> bool: function
553 bool can_use_flash_attention(sdp_params const& params, bool debug) { in can_use_flash_attention() function