Searched defs:sQ (Results 1 – 2 of 2) sorted by relevance
142 Tensor sQ = make_tensor(make_smem_ptr(reinterpret_cast<Element *>(smem_)), in compute_attn_1rowblock() local590 Tensor sQ = make_tensor(make_smem_ptr(reinterpret_cast<Element *>(smem_)), in compute_attn_1rowblock_splitkv() local
154 Tensor sQ = make_tensor(make_smem_ptr(reinterpret_cast<Element *>(smem_)), in compute_dq_dk_dv_1colblock() local