Searched defs:row_idx (Results 1 – 14 of 14) sorted by relevance
51 const int row_idx = row_idx_base + i * 8; in apply_mask_local() local170 const int row_idx = row_idx_base + i * 8; in apply_mask() local
57 const int row_idx = row_idx_base + i * 8; in apply_alibi() local
105 for (int row_idx = 0; row_idx < num_mi_rows; row_idx++) { in free_cdef_row_sync() local180 for (int row_idx = 0; row_idx < num_mi_rows; row_idx++) { in alloc_cdef_row_sync() local
101 for (int row_idx = 0; row_idx < num_rows; ++row_idx) { in Compute() local
104 for (int row_idx = 0; row_idx < num_rows; ++row_idx) { in Compute() local
237 for (int row_idx = 0; row_idx < num_rows; ++row_idx) { in operator ()() local
277 for (int64_t row_idx = 0; row_idx <= num_shard_rows; ++row_idx) { in GetSparseMatrixRef() local
349 xla::XlaOp row_idx = values[0]; in operator ()() local370 auto row_idx = values[0]; in operator ()() local
1570 const int row_idx = mv_max + row_val; in rd_update_mvcount() local1586 const int row_idx = mv_max + row_val; in rd_update_mvcount() local
386 const int row_idx = mv_max + row_val; in update_mvcount() local
984 let row_idx = i * self.cols; in push_col() localVariable1091 let row_idx = i * (self.cols - 1); in pop_col() localVariable1127 let row_idx = col_index + i * (self.cols - 1); in remove_col() localVariable
1890 unsigned row_idx = 0; in emit_inline_matrix_constructor() local
113 for (int row_idx = 1; row_idx < buf->crop_heights[is_uv]; row_idx++) { in set_planes_to_neutral_grey() local122 for (int row_idx = 0; row_idx < buf->crop_heights[is_uv]; row_idx++) { in set_planes_to_neutral_grey() local
3798 int32_t row_idx = -1; in ntq_emit_intrinsic() local