Searched defs:seqlen_knew (Results 1 – 2 of 2) sorted by relevance
69 …int b, seqlen_q, seqlen_k, seqlen_knew, d, seqlen_q_rounded, seqlen_k_rounded, d_rounded, rotary_d… member
1421 int seqlen_knew = k.size(1); in mha_fwd_kvcache() local