Searched refs:col_indices_vec (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | add_op.cc | 112 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()()
114 ConstCSRComponent<T> b_comp{b.row_pointers_vec(i), b.col_indices_vec(i), in operator ()()
137 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()()
139 ConstCSRComponent<T> b_comp{b.row_pointers_vec(i), b.col_indices_vec(i), in operator ()()
167 ConstCSRComponent<T> a_comp{a.row_pointers_vec(i), a.col_indices_vec(i), in operator ()()
169 ConstCSRComponent<T> b_comp{b.row_pointers_vec(i), b.col_indices_vec(i), in operator ()()
171 CSRComponent<T> c_comp{c->row_pointers_vec(i), c->col_indices_vec(i), in operator ()()
|
| D | sparse_ordering_amd_op.cc | 109 input_matrix->col_indices_vec(batch_index).data(), in Compute()
110 input_matrix->col_indices_vec(batch_index).data()); in Compute()
|
| D | sparse_mat_mul_op.cc | 275 csr_matrix.col_indices_vec(batch_index).data(), in GetSparseMatrixRef()
425 a_input_matrix->col_indices_vec(i), in Compute()
429 b_input_matrix->col_indices_vec(i), in Compute()
454 a_input_matrix->col_indices_vec(i), in Compute()
458 b_input_matrix->col_indices_vec(i), in Compute()
489 a_input_matrix->col_indices_vec(i), in Compute()
493 b_input_matrix->col_indices_vec(i), in Compute()
496 CSRComponent<T> c_comp{c.row_pointers_vec(i), c.col_indices_vec(i), in Compute()
|
| D | transpose_op.cc | 198 input_matrix.row_pointers_vec(i), input_matrix.col_indices_vec(i), in operator ()()
201 output_matrix->row_pointers_vec(i), output_matrix->col_indices_vec(i), in operator ()()
|
| D | sparse_matrix.h | 237 inline TTypes<int32>::UnalignedVec col_indices_vec(int batch) { in col_indices_vec() function
246 inline TTypes<int32>::UnalignedConstVec col_indices_vec(int batch) const { in col_indices_vec() function
|
| D | mat_mul_op.cc | 281 csr_matrix.col_indices_vec(batch_index).data() + row_offset, in GetSparseMatrixRef()
575 auto a_col_ind = a_matrix->col_indices_vec(i); in Compute()
665 auto a_col_ind = a_input_matrix->col_indices_vec(i); in Compute()
|
| D | sparse_cholesky_op.cc | 120 input_matrix->col_indices_vec(batch_index).data(), in Compute()
|