Home
last modified time | relevance | path

Searched defs:coo_col_ind (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dcsr_sparse_matrix_to_sparse_tensor_op.cc168 auto coo_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute() local
Dcsr_sparse_matrix_to_dense_op.cc162 auto coo_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute() local
Dsparse_tensor_to_csr_sparse_matrix_op.cc227 auto coo_col_ind = coo_col_ind_t.vec<int32>(); in ComputeAsync() local
Ddense_to_csr_sparse_matrix_op.cc294 auto coo_col_ind = coo_col_ind_t.vec<int32>(); in ComputeAsync() local
Dkernels_gpu.cu.cc150 TTypes<int>::Vec coo_col_ind) { in operator ()()
220 TTypes<int>::ConstVec coo_col_ind, TTypes<int64>::Matrix indices) { in operator ()()