Searched defs:lse_data (Results 1 – 1 of 1) sorted by relevance
267 accum_t* lse_data = logsumexp.data_ptr<accum_t>(); in cpu_flash_attention() local530 const accum_t* lse_data = logsumexp.const_data_ptr<accum_t>(); in cpu_flash_attention_backward() local