Searched refs:csr_matrix (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 264 const CSRSparseMatrix& csr_matrix, const int batch_index, in GetSparseMatrixRef() argument
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()
277 csr_matrix.dense_shape().vec<int64>()(csr_matrix.dims() - 1); in GetSparseMatrixRef()
281 csr_matrix.col_indices_vec(batch_index).data() + row_offset, in GetSparseMatrixRef()
282 csr_matrix.values_vec<T>(batch_index).data() + row_offset); in GetSparseMatrixRef()
|
| D | sparse_mat_mul_op.cc | 266 const CSRSparseMatrix& csr_matrix, const int rank, const int batch_index, in GetSparseMatrixRef() argument
268 const auto dense_shape = csr_matrix.dense_shape().vec<int64>(); in GetSparseMatrixRef()
273 num_rows, num_cols, csr_matrix.nnz(batch_index), in GetSparseMatrixRef()
274 csr_matrix.row_pointers_vec(batch_index).data(), in GetSparseMatrixRef()
275 csr_matrix.col_indices_vec(batch_index).data(), in GetSparseMatrixRef()
276 csr_matrix.values_vec<T>(batch_index).data()); in GetSparseMatrixRef()
|
| /external/eigen/bench/ |
| D | sparse_trisolver.cpp | 139 gmm::csr_matrix<Scalar> m2; in main()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 259 csr_mats = [list(map(sparse.csr_matrix, mat)) for mat in mats]
390 csr_mats = [[sparse.csr_matrix(m) for m in mat] for mat in mats]
|
| /external/python/pybind11/docs/advanced/cast/ |
| D | eigen.rst | 22 ``scipy.sparse.csr_matrix``/``scipy.sparse.csc_matrix`` objects.
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 102 // (http://scipy-lectures.org/advanced/scipy_sparse/csr_matrix.html)
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v3a.fbs | 105 // (http://scipy-lectures.org/advanced/scipy_sparse/csr_matrix.html)
|
| D | schema.fbs | 109 // (http://scipy-lectures.org/advanced/scipy_sparse/csr_matrix.html)
|