Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/
Dattention.cu1084 constexpr int kAlignLSE = 1; in _efficient_attention_forward() local
1176 constexpr decltype(M) kAlignLSE = Kernel::kAlignLSE; in _efficient_attention_forward() local
/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dkernel_forward.h113 static constexpr int32_t kAlignLSE = 32; // block size of backward member