Searched defs:tVgV (Results 1 – 2 of 2) sorted by relevance
158 Tensor tVgV = gmem_thr_copy_QKV.partition_S(gV); // (VCPY, VCPY_N, VCPY_K, nblocksN) in compute_attn_1rowblock() local604 Tensor tVgV = gmem_thr_copy_QKV.partition_S(gV); // (VCPY, VCPY_N, VCPY_K) in compute_attn_1rowblock_splitkv() local
203 Tensor tVgV = gmem_thr_copy_QKV.partition_S(gV); // (VCPY, VCPY_N, VCPY_K) in compute_dq_dk_dv_1colblock() local