Searched defs:row_idx_base (Results 1 – 2 of 2) sorted by relevance
48 const int row_idx_base = row_idx_offset + mi * warp_row_stride; in apply_mask_local() local167 const int row_idx_base = row_idx_offset + mi * warp_row_stride; in apply_mask() local
54 const int row_idx_base = row_idx_offset + mi * warp_row_stride; in apply_alibi() local