Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
Dmma_accum_lambda_iterator.h42 int lane_in_quad = (lane_id & 3); in get_lane_offset() local
90 int lane_in_quad = (lane_id & 3); in reduceSameRow() local
122 int lane_in_quad = (lane_id & 3); in get_lane_offset() local