Searched defs:lse_dim (Results 1 – 1 of 1) sorted by relevance
203 auto lse_dim = ceil_div((int32_t)num_queries, kAlignLSE) * kAlignLSE; in advance_to_block() local1166 auto lse_dim = ceil_div((int32_t)p.num_queries, kAlignLSE) * kAlignLSE; in attention_kernel() local