Searched defs:kvSplitSize (Results 1 – 2 of 2) sorted by relevance
242 int64_t kvSplitSize = kv_split_size > kvSize ? kvSize : kv_split_size; in cpu_flash_attention() local498 int64_t kvSplitSize = kv_split_size > kvSize ? kvSize : kv_split_size; in cpu_flash_attention_backward() local
365 int64_t kvSplitSize = kv_split_size > kvSize ? kvSize : kv_split_size; in cpu_flash_attention() local