Home
last modified time | relevance | path

Searched refs:compute_attn_1rowblock_splitkv (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dflash_fwd_kernel.h479 inline __device__ void compute_attn_1rowblock_splitkv(const Params &params, const int bidb, const i… in compute_attn_1rowblock_splitkv() function
1075 …pytorch_flash::compute_attn_1rowblock_splitkv<Kernel_traits, Is_causal, Is_local, Has_alibi, Is_ev… in compute_attn_splitkv()