Searched refs:col_indices (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_matrix.h | 149 const Tensor& col_indices, in CreateCSRSparseMatrix() argument
153 col_indices, values); in CreateCSRSparseMatrix()
285 inline Tensor& col_indices() { in col_indices() function
290 inline const Tensor& col_indices() const { in col_indices() function
346 Tensor col_indices(p.tensors_[3]); in Decode()
351 row_pointers, col_indices, values); in Decode()
359 col_indices_ = std::move(col_indices); in Decode()
391 TF_RETURN_IF_ERROR(copy(from.col_indices(), &to_col_ind)); in DeviceCopy()
402 const Tensor& col_indices, const Tensor& values) in CSRSparseMatrix() argument
407 col_indices_(col_indices),
[all …]
|
| D | conj_op.cc | 57 a.row_pointers(), a.col_indices(), b_values_t, b)); in operator ()()
79 a.row_pointers(), a.col_indices(), a.values(), b)); \
|
| D | sparse_matrix_components_op.cc | 67 c->set_output(1, csr_sparse_matrix->col_indices()); in Compute()
97 /*input*/ csr_sparse_matrix->col_indices().vec<int32>(), in Compute()
|
| D | softmax_op.cc | 73 logits_matrix->col_indices(), output_values_t, &output_matrix)); in Compute()
180 softmax_matrix->row_pointers(), softmax_matrix->col_indices(), in Compute()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 95 auto csr_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute()
168 auto coo_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute()
|
| D | csr_sparse_matrix_to_dense_op.cc | 78 auto col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute()
162 auto coo_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute()
|
| D | mul_op.cc | 136 a.row_pointers(), a.col_indices(), c_values_t, c)); in Compute()
|
| D | kernels_gpu.cu.cc | 295 a.row_pointers(), a.col_indices(), c_values_t, c)); in CSRSparseMatrixBatchMulVecImpl()
614 auto softmax_col_ind = softmax.col_indices().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl()
618 auto grad_softmax_col_ind = grad_softmax.col_indices().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl()
|
| /external/eigen/doc/examples/ |
| D | nullary_indexing.cpp | 20 …exing_functor(const ArgType& arg, const RowIndexType& row_indices, const ColIndexType& col_indices) in indexing_functor() argument
21 : m_arg(arg), m_rowIndices(row_indices), m_colIndices(col_indices) in indexing_functor()
33 … Eigen::MatrixBase<ArgType>& arg, const RowIndexType& row_indices, const ColIndexType& col_indices) in indexing() argument
37 …MatrixType::NullaryExpr(row_indices.size(), col_indices.size(), Func(arg.derived(), row_indices, c… in indexing()
|