Home
last modified time | relevance | path

Searched defs:accum_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.h60 int accum_m = mma_m * InstructionShape::kM * OpDelta::kRow + in iterateRows() local
123 int accum_m, accum_n; in get_lane_offset() local
169 int accum_m = tile_m * Policy::InterleavedTile::kRow + in iterateRows() local
240 int accum_m = mma_m * Delta::kRow + m + lane_offset.row(); in iterateRows() local