Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc127 TTypes<int>::UnalignedVec coo_row_ind) { in operator ()()
149 TTypes<int64>::ConstMatrix indices, TTypes<int>::Vec coo_row_ind, in operator ()()
219 TTypes<int>::ConstVec host_batch_ptr, TTypes<int>::Vec coo_row_ind, in operator ()()
Dcsr_sparse_matrix_to_sparse_tensor_op.cc174 auto coo_row_ind = coo_row_ind_t.vec<int32>(); in Compute() local
Dcsr_sparse_matrix_to_dense_op.cc168 auto coo_row_ind = coo_row_ind_t.vec<int32>(); in Compute() local
Dsparse_tensor_to_csr_sparse_matrix_op.cc226 auto coo_row_ind = coo_row_ind_t.vec<int32>(); in ComputeAsync() local
Ddense_to_csr_sparse_matrix_op.cc293 auto coo_row_ind = coo_row_ind_t.vec<int32>(); in ComputeAsync() local