Searched defs:qSplitSize (Results 1 – 2 of 2) sorted by relevance
241 int64_t qSplitSize = q_split_size > qSize ? qSize : q_split_size; in cpu_flash_attention() local497 int64_t qSplitSize = q_split_size > qSize ? qSize : q_split_size; in cpu_flash_attention_backward() local
364 int64_t qSplitSize = q_split_size > qSize ? qSize : q_split_size; in cpu_flash_attention() local