Home
last modified time | relevance | path

Searched defs:kMaxK (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
Dmma_from_smem.h152 static constexpr int kMaxK = kMaxK_; variable
/external/zstd/lib/dictBuilder/
Dfastcover.c633 const unsigned kMaxK = parameters->k == 0 ? 2000 : parameters->k; in ZDICT_optimizeTrainFromBuffer_fastCover() local
Dcover.c1179 const unsigned kMaxK = parameters->k == 0 ? 2000 : parameters->k; local
/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dkernel_forward.h110 static constexpr int kMaxK = kMaxK_; member
Dkernel_backward.h241 static constexpr int kMaxK = kMaxK_; member