Searched defs:row_offset_oaccum (Results 1 – 1 of 1) sorted by relevance
523 …const index_t row_offset_oaccum = (((n_split_idx * params.b + bidb) * params.h + bidh) * params.se… in compute_attn_1rowblock_splitkv() local993 …const index_t row_offset_oaccum = (((n_split_idx * params.b + bidb) * params.h + bidh) * params.se… in compute_attn_1rowblock_splitkv() local1178 const index_t row_offset_oaccum = bidx * kBlockM * params.d_rounded; in combine_attn_seqk_parallel() local