Searched refs:coo_col_ind (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 163 TTypes<int>::Vec coo_col_ind) { in operator ()() argument
173 coo_col_ind.data(), size)); in operator ()()
178 coo_col_ind.data(), size)); in operator ()()
233 TTypes<int>::ConstVec coo_col_ind, TTypes<int64>::Matrix indices) { in operator ()() argument
238 DCHECK_NE(coo_col_ind.data(), nullptr); in operator ()()
242 DCHECK_EQ(size, coo_col_ind.size()); in operator ()()
249 coo_col_ind.data(), indices.data(), size)); in operator ()()
265 coo_row_ind.data(), coo_col_ind.data(), indices.data(), in operator ()()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 168 auto coo_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute() local
194 coo_col_ind, indices)); in Compute()
230 TTypes<int>::ConstVec coo_col_ind,
|
| D | csr_sparse_matrix_to_dense_op.cc | 162 auto coo_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute() local
188 coo_col_ind, indices)); in Compute()
246 TTypes<int>::ConstVec coo_col_ind,
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 226 auto coo_col_ind = coo_col_ind_t.vec<int32>(); in ComputeAsync() local
232 st_to_coo(d, dense_shape, indices, coo_row_ind, coo_col_ind); in ComputeAsync()
301 TTypes<int>::Vec coo_row_ind, TTypes<int>::Vec coo_col_ind);
|
| D | kernels.h | 62 TTypes<int32>::Vec coo_col_ind);
89 TTypes<int32>::ConstVec coo_col_ind,
|
| D | dense_to_csr_sparse_matrix_op.cc | 293 auto coo_col_ind = coo_col_ind_t.vec<int32>(); in ComputeAsync() local
299 st_to_coo(d, dense_shape, indices, coo_row_ind, coo_col_ind); in ComputeAsync()
381 TTypes<int>::Vec coo_row_ind, TTypes<int>::Vec coo_col_ind);
|