Home
last modified time | relevance | path

Searched refs:cuda_sparse (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dtranspose_op.cc263 GpuSparse cuda_sparse(ctx); in operator ()() local
264 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
281 return cuda_sparse.Csr2csc( in operator ()()
Dsparse_tensor_to_csr_sparse_matrix_op.cc310 GpuSparse cuda_sparse(c); in operator ()() local
311 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
312 return cuda_sparse.Coo2csr(coo_row_ind.data(), in operator ()()
Dmat_mul_op.cc728 GpuSparse cuda_sparse(ctx); in Compute() local
729 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in Compute()
800 cuda_sparse.Csrmm(transA, transB, m, n, k, nnz, &alpha, descrA, in Compute()
822 GpuSparse cuda_sparse(ctx); in Compute() local
823 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in Compute()
854 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, descrA, in Compute()
Ddense_to_csr_sparse_matrix_op.cc390 GpuSparse cuda_sparse(c); in operator ()() local
391 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()()
392 return cuda_sparse.Coo2csr(coo_row_ind.data(), in operator ()()
DBUILD82 "//tensorflow/core/kernels:cuda_sparse",
/external/tensorflow/tensorflow/core/kernels/
DBUILD3535 name = "cuda_sparse",
3536 srcs = if_cuda(["cuda_sparse.cc"]) + if_rocm(["rocm_sparse.cc"]),
3537 hdrs = ["cuda_sparse.h"],
3671 ":cuda_sparse",
3680 ":cuda_sparse",