Searched defs:qBlockSize (Results 1 – 2 of 2) sorted by relevance
285 int64_t qBlockSize = std::min(qSplitSize, qSize - m); in cpu_flash_attention() local551 int64_t qBlockSize = std::min(qSplitSize, qSize - m); in cpu_flash_attention_backward() local
422 int64_t qBlockSize = std::min(qSplitSize, qSize - m); in cpu_flash_attention() local