Home
last modified time | relevance | path

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

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