Searched defs:ompIdx (Results 1 – 2 of 2) sorted by relevance
274 int ompIdx = at::get_thread_num(); in cpu_flash_attention() local537 int ompIdx = at::get_thread_num(); in cpu_flash_attention_backward() local
410 int ompIdx = torch::executor::get_thread_num(); in cpu_flash_attention() local