Home
last modified time | relevance | path

Searched defs:cuda_sparse (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dtranspose_op.cc263 GpuSparse cuda_sparse(ctx); in operator ()() local
Dsparse_tensor_to_csr_sparse_matrix_op.cc311 GpuSparse cuda_sparse(c); in operator ()() local
Ddense_to_csr_sparse_matrix_op.cc389 GpuSparse cuda_sparse(c); in operator ()() local
Dmat_mul_op.cc781 GpuSparse cuda_sparse(ctx); in Compute() local
921 GpuSparse cuda_sparse(ctx); in Compute() local