Searched defs:csr_row_ptr (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_test.cc | 34 auto csr_row_ptr = test::AsTensor<int32>({0, 0, 0, 0, 0}); in TEST() local
58 Tensor csr_row_ptr(DT_INT32, {12}); in TEST() local
|
| D | kernels.cc | 32 TTypes<int32>::Vec csr_row_ptr, TTypes<int32>::Vec csr_col_ind) { in operator ()()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 94 auto csr_row_ptr = csr_sparse_matrix->row_pointers().vec<int32>(); in Compute() local
167 auto csr_row_ptr = csr_sparse_matrix->row_pointers().vec<int32>(); in Compute() local
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 81 Tensor csr_row_ptr(cpu_allocator(), DT_INT32, in Compute() local
228 auto csr_row_ptr = csr_row_ptr_t.vec<int32>(); in ComputeAsync() local
|
| D | dense_to_csr_sparse_matrix_op.cc | 105 Tensor csr_row_ptr(cpu_allocator(), DT_INT32, in Compute() local
295 auto csr_row_ptr = csr_row_ptr_t.vec<int32>(); in ComputeAsync() local
|
| D | csr_sparse_matrix_to_dense_op.cc | 161 auto csr_row_ptr = csr_sparse_matrix->row_pointers().vec<int32>(); in Compute() local
|
| D | kernels_gpu.cu.cc | 126 OpKernelContext* c, TTypes<const int>::UnalignedVec csr_row_ptr, in operator ()()
|