Home
last modified time | relevance | path

Searched defs:qSplitSize (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DFlashAttentionKernel.cpp241 int64_t qSplitSize = q_split_size > qSize ? qSize : q_split_size; in cpu_flash_attention() local
497 int64_t qSplitSize = q_split_size > qSize ? qSize : q_split_size; in cpu_flash_attention_backward() local
/external/executorch/extension/llm/custom_ops/
Dop_sdpa.cpp364 int64_t qSplitSize = q_split_size > qSize ? qSize : q_split_size; in cpu_flash_attention() local