Searched refs:row_pointers_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_mat_mul_op.cc | 274 csr_matrix.row_pointers_vec(batch_index).data(), in GetSparseMatrixRef()
424 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()
428 ConstCSRComponent<T> b_comp{b_input_matrix->row_pointers_vec(i), in Compute()
453 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()
457 ConstCSRComponent<T> b_comp{b_input_matrix->row_pointers_vec(i), in Compute()
488 ConstCSRComponent<T> a_comp{a_input_matrix->row_pointers_vec(i), in Compute()
492 ConstCSRComponent<T> b_comp{b_input_matrix->row_pointers_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 | 195 input_matrix.row_pointers_vec(i), input_matrix.col_indices_vec(i), in operator ()()
198 output_matrix->row_pointers_vec(i), output_matrix->col_indices_vec(i), in operator ()()
|
| D | sparse_ordering_amd_op.cc | 108 input_matrix->row_pointers_vec(batch_index).data(), in Compute()
|
| D | mat_mul_op.cc | 270 csr_matrix.row_pointers_vec(batch_index)(row_begin); in GetSparseMatrixRef()
273 csr_matrix.row_pointers_vec(batch_index)(row_begin + row_idx) - in GetSparseMatrixRef()
577 auto a_row_ptr = a_matrix->row_pointers_vec(i); in Compute()
667 auto a_row_ptr = a_input_matrix->row_pointers_vec(i); in Compute()
|
| D | sparse_matrix.h | 219 inline TTypes<int32>::UnalignedVec row_pointers_vec(int batch) { in row_pointers_vec() function
228 inline TTypes<int32>::UnalignedConstVec row_pointers_vec(int batch) const { in row_pointers_vec() function
|
| D | sparse_cholesky_op.cc | 121 input_matrix->row_pointers_vec(batch_index).data(), in Compute()
|