Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
Dmma_accum_lambda_iterator.h57 for (int mma_m = 0; mma_m < Policy::MmaIterations::kRow; ++mma_m) { in iterateRows() local
166 for (int mma_m = 0; mma_m < Policy::MmaIterations::kRow; ++mma_m) { in iterateRows() local
237 for (int mma_m = 0; mma_m < Iterations::kRow; ++mma_m) { in iterateRows() local