Searched defs:kvSize (Results 1 – 3 of 3) sorted by relevance
153 int64_t kvSize) { in reshape_attn_mask_to_4d()205 int64_t kvSize = value.size(1); in cpu_flash_attention() local448 int64_t kvSize = value.size(1); in cpu_flash_attention_backward() local
288 auto kvSize = params.key.sym_size(2); in check_attn_mask_shape() local
264 int64_t kvSize = value.size(2); in cpu_flash_attention() local