Searched defs:Has_alibi (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | flash_fwd_launch_template.h | 35 …EL(flash_fwd_kernel, bool Is_dropout, bool Is_causal, bool Is_local, bool Has_alibi, bool Is_even_… in DEFINE_FLASH_FORWARD_KERNEL()
44 DEFINE_FLASH_FORWARD_KERNEL(flash_fwd_splitkv_kernel, bool Is_causal, bool Is_local, bool Has_alibi… in DEFINE_FLASH_FORWARD_KERNEL()
|
| D | flash_bwd_launch_template.h | 36 …NEL(flash_bwd_dq_dk_dv_loop_kernel, bool Is_dropout, bool Is_causal, bool Has_alibi, bool Is_even_… in DEFINE_FLASH_BACKWARD_KERNEL()
44 …eqk_parallel_kernel, bool Is_dropout, bool Is_causal, bool Is_local, bool Has_alibi, bool Is_even_… in DEFINE_FLASH_BACKWARD_KERNEL()
|