Searched defs:has_attn_mask (Results 1 – 2 of 2) sorted by relevance
209 bool has_attn_mask = attn_mask.has_value() && attn_mask.value().numel(); in cpu_flash_attention() local452 bool has_attn_mask = attn_mask.has_value() && attn_mask.value().numel(); in cpu_flash_attention_backward() local
288 bool has_attn_mask = attn_mask.has_value() && attn_mask.value().numel(); in cpu_flash_attention() local