Searched defs:kAlignLSE (Results 1 – 2 of 2) sorted by relevance
1084 constexpr int kAlignLSE = 1; in _efficient_attention_forward() local1176 constexpr decltype(M) kAlignLSE = Kernel::kAlignLSE; in _efficient_attention_forward() local
113 static constexpr int32_t kAlignLSE = 32; // block size of backward member