Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dflash_bwd_launch_template.h36 DEFINE_FLASH_BACKWARD_KERNEL(flash_bwd_dq_dk_dv_loop_kernel, bool Is_dropout, bool Is_causal, bool … in DEFINE_FLASH_BACKWARD_KERNEL()
44 DEFINE_FLASH_BACKWARD_KERNEL(flash_bwd_dq_dk_dv_loop_seqk_parallel_kernel, bool Is_dropout, bool Is… in DEFINE_FLASH_BACKWARD_KERNEL()
Dflash_fwd_launch_template.h35 DEFINE_FLASH_FORWARD_KERNEL(flash_fwd_kernel, bool Is_dropout, bool Is_causal, bool Is_local, bool … in DEFINE_FLASH_FORWARD_KERNEL()