Searched refs:cuda_sparse (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | transpose_op.cc | 263 GpuSparse cuda_sparse(ctx); in operator ()() local 264 TF_RETURN_IF_ERROR(cuda_sparse.Initialize()); in operator ()() 281 return cuda_sparse.Csr2csc( in operator ()()
|
D | sparse_tensor_to_csr_sparse_matrix_op.cc | 310 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 ()()
|
D | mat_mul_op.cc | 728 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()
|
D | dense_to_csr_sparse_matrix_op.cc | 390 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 ()()
|
D | BUILD | 82 "//tensorflow/core/kernels:cuda_sparse",
|
/external/tensorflow/tensorflow/core/kernels/ |
D | BUILD | 3535 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",
|