Home
last modified time | relevance | path

Searched defs:gSin (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Drotary.h121 …Tensor gSin = make_tensor(Sin(_, m, k).data() + (is_left ? 0 : -rotary_dim / 2), Sin(_, m, k).layo… in copy_rotary_contiguous() local
Dflash_fwd_kernel.h673 …Tensor gSin = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.rotary_sin_ptr) + row_o… in compute_attn_1rowblock_splitkv() local
776 …Tensor gSin = make_tensor(make_gmem_ptr(reinterpret_cast<Element *>(params.rotary_sin_ptr) + row_o… in compute_attn_1rowblock_splitkv() local