Searched defs:csr_col_ind (Results 1 – 5 of 5) sorted by relevance
33 Tensor csr_col_ind(DT_INT32, {4}); in TEST() local56 Tensor csr_col_ind(DT_INT32, {3}); in TEST() local
32 TTypes<int32>::Vec csr_row_ptr, TTypes<int32>::Vec csr_col_ind) { in operator ()()
95 auto csr_col_ind = csr_sparse_matrix->col_indices().vec<int32>(); in Compute() local
80 Tensor csr_col_ind(cpu_allocator(), DT_INT32, TensorShape({total_nnz})); in Compute() local
104 Tensor csr_col_ind(cpu_allocator(), DT_INT32, TensorShape({total_nnz})); in Compute() local