Searched refs:cuda_sparse (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 781 GpuSparse cuda_sparse(ctx); in Compute() local
782 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in Compute()
854 TF_RETURN_IF_ERROR(cuda_sparse.SpMMBufferSize( in Compute()
863 TF_RETURN_IF_ERROR(cuda_sparse.SpMM(transA, transB, &alpha, matA, matB, in Compute()
897 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
921 GpuSparse cuda_sparse(ctx); in Compute() local
922 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in Compute()
955 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, in Compute()
959 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, descrA, in Compute()
|
| D | transpose_op.cc | 263 GpuSparse cuda_sparse(ctx); in operator ()() local
264 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
277 return cuda_sparse.Csr2csc( in operator ()()
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 311 GpuSparse cuda_sparse(c); in operator ()() local
312 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
313 return cuda_sparse.Coo2csr(coo_row_ind.data(), in operator ()()
|
| D | dense_to_csr_sparse_matrix_op.cc | 389 GpuSparse cuda_sparse(c); in operator ()() local
390 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
391 return cuda_sparse.Coo2csr(coo_row_ind.data(), in operator ()()
|
| D | BUILD | 85 "//tensorflow/core/util:cuda_sparse",
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | BUILD | 237 "//tensorflow/core/util:cuda_sparse",
246 "//tensorflow/core/util:cuda_sparse",
|
| /external/tensorflow/tensorflow/core/util/ |
| D | BUILD | 646 name = "cuda_sparse",
647 srcs = if_cuda(["cuda_sparse.cc"]) + if_rocm(["rocm_sparse.cc"]),
648 hdrs = ["cuda_sparse.h"],
|