Searched defs:Is_causal (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 DEFINE_FLASH_FORWARD_KERNEL(flash_fwd_kernel, bool Is_dropout, bool Is_causal, bool Is_local, bool … 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 DEFINE_FLASH_BACKWARD_KERNEL(flash_bwd_dq_dk_dv_loop_kernel, bool Is_dropout, bool Is_causal, bool … in DEFINE_FLASH_BACKWARD_KERNEL()
44 …ERNEL(flash_bwd_dq_dk_dv_loop_seqk_parallel_kernel, bool Is_dropout, bool Is_causal, bool Is_local… in DEFINE_FLASH_BACKWARD_KERNEL()
|