Lines Matching refs:bs

81 Init(int num_eliminate_blocks, const CompressedRowBlockStructure* bs) {  in Init()  argument
88 const int num_col_blocks = bs->cols.size(); in Init()
89 const int num_row_blocks = bs->rows.size(); in Init()
102 lhs_num_rows += bs->cols[i].size; in Init()
112 const int chunk_block_id = bs->rows[r].cells.front().block_id; in Init()
122 const int e_block_size = bs->cols[chunk_block_id].size; in Init()
127 const CompressedRow& row = bs->rows[r + chunk.size]; in Init()
138 buffer_size += e_block_size * bs->cols[cell.block_id].size; in Init()
183 const CompressedRowBlockStructure* bs = A->block_structure(); in Eliminate() local
184 const int num_col_blocks = bs->cols.size(); in Eliminate()
196 const int block_size = bs->cols[i].size; in Eliminate()
198 diag(D + bs->cols[i].position, block_size); in Eliminate()
231 const int e_block_id = bs->rows[chunk.start].cells.front().block_id; in Eliminate()
232 const int e_block_size = bs->cols[e_block_id].size; in Eliminate()
241 diag(D + bs->cols[e_block_id].position, e_block_size); in Eliminate()
293 ChunkOuterProduct(bs, inverse_ete, buffer, chunk.buffer_layout, lhs); in Eliminate()
309 const CompressedRowBlockStructure* bs = A->block_structure(); in BackSubstitute() local
313 const int e_block_id = bs->rows[chunk.start].cells.front().block_id; in BackSubstitute()
314 const int e_block_size = bs->cols[e_block_id].size; in BackSubstitute()
316 double* y_ptr = y + bs->cols[e_block_id].position; in BackSubstitute()
323 diag(D + bs->cols[e_block_id].position, e_block_size); in BackSubstitute()
331 const CompressedRow& row = bs->rows[chunk.start + j]; in BackSubstitute()
339 (b + bs->rows[chunk.start + j].block.position, row.block.size); in BackSubstitute()
343 const int f_block_size = bs->cols[f_block_id].size; in BackSubstitute()
380 const CompressedRowBlockStructure* bs = A->block_structure(); in UpdateRhs() local
381 const int e_block_id = bs->rows[chunk.start].cells.front().block_id; in UpdateRhs()
382 const int e_block_size = bs->cols[e_block_id].size; in UpdateRhs()
384 int b_pos = bs->rows[row_block_counter].block.position; in UpdateRhs()
387 const CompressedRow& row = bs->rows[row_block_counter + j]; in UpdateRhs()
400 const int block_size = bs->cols[block_id].size; in UpdateRhs()
443 const CompressedRowBlockStructure* bs = A->block_structure(); in ChunkDiagonalBlockAndGradient() local
445 int b_pos = bs->rows[row_block_counter].block.position; in ChunkDiagonalBlockAndGradient()
454 const CompressedRow& row = bs->rows[row_block_counter + j]; in ChunkDiagonalBlockAndGradient()
479 const int f_block_size = bs->cols[f_block_id].size; in ChunkDiagonalBlockAndGradient()
499 ChunkOuterProduct(const CompressedRowBlockStructure* bs, in ChunkOuterProduct() argument
522 const int block1_size = bs->cols[it1->first].size; in ChunkOuterProduct()
538 const int block2_size = bs->cols[it2->first].size; in ChunkOuterProduct()
561 const CompressedRowBlockStructure* bs = A->block_structure(); in NoEBlockRowsUpdate() local
563 for (; row_block_counter < bs->rows.size(); ++row_block_counter) { in NoEBlockRowsUpdate()
564 const CompressedRow& row = bs->rows[row_block_counter]; in NoEBlockRowsUpdate()
567 const int block_size = bs->cols[block_id].size; in NoEBlockRowsUpdate()
599 const CompressedRowBlockStructure* bs = A->block_structure(); in NoEBlockRowOuterProduct() local
600 const CompressedRow& row = bs->rows[row_block_index]; in NoEBlockRowOuterProduct()
606 const int block1_size = bs->cols[row.cells[i].block_id].size; in NoEBlockRowOuterProduct()
631 const int block2_size = bs->cols[row.cells[j].block_id].size; in NoEBlockRowOuterProduct()
652 const CompressedRowBlockStructure* bs = A->block_structure(); in EBlockRowOuterProduct() local
653 const CompressedRow& row = bs->rows[row_block_index]; in EBlockRowOuterProduct()
659 const int block1_size = bs->cols[row.cells[i].block_id].size; in EBlockRowOuterProduct()
678 const int block2_size = bs->cols[row.cells[j].block_id].size; in EBlockRowOuterProduct()