Searched defs:gCos (Results 1 – 2 of 2) sorted by relevance
120 …Tensor gCos = make_tensor(Cos(_, m, k).data() + (is_left ? 0 : -rotary_dim / 2), Cos(_, m, k).layo… in copy_rotary_contiguous() local
670 …Tensor gCos = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.rotary_cos_ptr) + row_o… in compute_attn_1rowblock_splitkv() local773 …Tensor gCos = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.rotary_cos_ptr) + row_o… in compute_attn_1rowblock_splitkv() local