Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
Dmma_accum_lambda_iterator.h70 int accum_n = mma_n * InstructionShape::kN * OpDelta::kColumn + in iterateRows() local
123 int accum_m, accum_n; in get_lane_offset() local
190 int accum_n = tile_n * Policy::InterleavedTile::kColumn + in iterateRows() local
245 int accum_n = in iterateRows() local