Searched refs:row_ptrs (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 266 std::vector<int32>* row_ptrs) { in GetSparseMatrixRef() argument
268 row_ptrs->resize(num_shard_rows + 1); in GetSparseMatrixRef()
272 row_ptrs->at(row_idx) = in GetSparseMatrixRef()
280 row_ptrs->at(num_shard_rows) /* total_nnz */, row_ptrs->data(), in GetSparseMatrixRef()
312 std::vector<int32> row_ptrs; in SparseDenseMatMulWithoutTransposedLHS() local
314 lhs, batch_idx, row_begin, num_shard_rows, &row_ptrs); in SparseDenseMatMulWithoutTransposedLHS()
383 std::vector<int32> row_ptrs; in SparseDenseMatMulWithTransposedLHS() local
385 lhs, batch_idx, row_begin, num_shard_rows, &row_ptrs); in SparseDenseMatMulWithTransposedLHS()
|
| D | sparse_matrix_components_op.cc | 81 auto row_ptrs = row_ptrs_t->vec<int32>(); in Compute() local
90 /*output*/ row_ptrs, in Compute()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CSRSparseMatrixComponents.pbtxt | 12 name: "row_ptrs"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CSRSparseMatrixComponents.pbtxt | 12 name: "row_ptrs"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CSRSparseMatrixComponents.pbtxt | 13 name: "row_ptrs"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 192 ShapeHandle row_ptrs = c->Vector(row_ptrs_dh); in __anon5631e9dc0502() local
193 c->set_output(0, row_ptrs); in __anon5631e9dc0502()
|
| D | ops.pbtxt | 6566 name: "row_ptrs"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 234 self.assertAllEqual(a_csr_mat.indptr, a_sm_val.row_ptrs)
290 self.assertAllEqual(csr_mat.indptr, sm_val.row_ptrs)
366 self.assertAllEqual(a_csr_mat.indptr, a_sm_val.row_ptrs)
415 self.assertAllEqual(csr_mat.indptr, sm_val.row_ptrs)
859 self.assertAllEqual(component.row_ptrs, np.zeros(rows + 1, np.int32))
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 38734 …cope, csr_sparse_matrix tf.Output, index tf.Output, type_ tf.DataType) (row_ptrs tf.Output, col_in…
|