Lines Matching refs:matrix

58   CompressedRowSparseMatrix matrix(1, 1, 1);
59 matrix.mutable_rows()[0] = 0;
60 matrix.mutable_rows()[1] = 1;
61 matrix.mutable_cols()[0] = 0;
62 matrix.mutable_values()[0] = 2;
65 IncompleteLQFactorization(matrix, 1, 0.0, 1, 0.0));
66 ExpectMatricesAreEqual(matrix, *l, 1e-16);
83 CompressedRowSparseMatrix matrix(10, 10, 100);
84 int* rows = matrix.mutable_rows();
85 int* cols = matrix.mutable_cols();
86 double* values = matrix.mutable_values();
103 IncompleteLQFactorization(matrix, 10, 0.0, 10, 0.0));
107 // Use Cholesky factorization to compute the L matrix.
121 // Allocate space and then make it a zero sized matrix.
122 CompressedRowSparseMatrix matrix(10, 10, 100);
123 matrix.set_num_rows(0);
136 DropEntriesAndAddRow(dense_vector, 1, 1, 0, &scratch, &matrix);
137 EXPECT_EQ(matrix.num_rows(), 1);
138 EXPECT_EQ(matrix.num_cols(), 10);
139 EXPECT_EQ(matrix.num_nonzeros(), 1);
140 EXPECT_EQ(matrix.values()[0], 5.0);
141 EXPECT_EQ(matrix.cols()[0], 0);
144 DropEntriesAndAddRow(dense_vector, 6, 10, 0, &scratch, &matrix);
145 EXPECT_EQ(matrix.num_rows(), 2);
146 EXPECT_EQ(matrix.num_cols(), 10);
147 EXPECT_EQ(matrix.num_nonzeros(), 7);
148 for (int idx = matrix.rows()[1]; idx < matrix.rows()[2]; ++idx) {
149 EXPECT_EQ(matrix.cols()[idx], idx - matrix.rows()[1]);
150 EXPECT_EQ(matrix.values()[idx], dense_vector(idx - matrix.rows()[1]));
154 DropEntriesAndAddRow(dense_vector, 6, 3, 0, &scratch, &matrix);
155 EXPECT_EQ(matrix.num_rows(), 3);
156 EXPECT_EQ(matrix.num_cols(), 10);
157 EXPECT_EQ(matrix.num_nonzeros(), 10);
159 EXPECT_EQ(matrix.cols()[matrix.rows()[2]], 0);
160 EXPECT_EQ(matrix.cols()[matrix.rows()[2] + 1], 3);
161 EXPECT_EQ(matrix.cols()[matrix.rows()[2] + 2], 5);
163 EXPECT_EQ(matrix.values()[matrix.rows()[2]], 5);
164 EXPECT_EQ(matrix.values()[matrix.rows()[2] + 1], 3);
165 EXPECT_EQ(matrix.values()[matrix.rows()[2] + 2], 4);
168 DropEntriesAndAddRow(dense_vector, 6, 6, 0.2, &scratch, &matrix);
169 EXPECT_EQ(matrix.num_rows(), 4);
170 EXPECT_EQ(matrix.num_cols(), 10);
171 EXPECT_EQ(matrix.num_nonzeros(), 14);
173 EXPECT_EQ(matrix.cols()[matrix.rows()[3]], 0);
174 EXPECT_EQ(matrix.cols()[matrix.rows()[3] + 1], 2);
175 EXPECT_EQ(matrix.cols()[matrix.rows()[3] + 2], 3);
176 EXPECT_EQ(matrix.cols()[matrix.rows()[3] + 3], 5);
178 EXPECT_EQ(matrix.values()[matrix.rows()[3]], 5);
179 EXPECT_EQ(matrix.values()[matrix.rows()[3] + 1], 2);
180 EXPECT_EQ(matrix.values()[matrix.rows()[3] + 2], 3);
181 EXPECT_EQ(matrix.values()[matrix.rows()[3] + 3], 4);
184 DropEntriesAndAddRow(dense_vector, 6, 2, 0.2, &scratch, &matrix);
185 EXPECT_EQ(matrix.num_rows(), 5);
186 EXPECT_EQ(matrix.num_cols(), 10);
187 EXPECT_EQ(matrix.num_nonzeros(), 16);
189 EXPECT_EQ(matrix.cols()[matrix.rows()[4]], 0);
190 EXPECT_EQ(matrix.cols()[matrix.rows()[4] + 1], 5);
192 EXPECT_EQ(matrix.values()[matrix.rows()[4]], 5);
193 EXPECT_EQ(matrix.values()[matrix.rows()[4] + 1], 4);