Searched refs:cuda_sparse (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 782 GpuSparse cuda_sparse(ctx); in Compute() local
783 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()
892 TF_RETURN_IF_ERROR(cuda_sparse.SpMMBufferSize( in Compute()
901 TF_RETURN_IF_ERROR(cuda_sparse.SpMM(transA, transB, &alpha, matA, matB, in Compute()
935 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
959 GpuSparse cuda_sparse(ctx); in Compute() local
960 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in Compute()
993 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, in Compute()
[all …]
|
| D | transpose_op.cc | 260 GpuSparse cuda_sparse(ctx); in operator ()() local
261 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
274 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 | 86 "//tensorflow/core/util:cuda_sparse",
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | BUILD | 240 "//tensorflow/core/util:cuda_sparse",
249 "//tensorflow/core/util:cuda_sparse",
|
| /external/tensorflow/tensorflow/core/util/ |
| D | BUILD | 695 name = "cuda_sparse",
696 srcs = if_cuda(["cuda_sparse.cc"]) + if_rocm(["rocm_sparse.cc"]),
697 hdrs = ["cuda_sparse.h"],
|