Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dmat_mul_op.cc788 GpuSparse cuda_sparse(ctx); in Compute() local
965 GpuSparse cuda_sparse(ctx); in Compute() local
H A Dtranspose_op.cc260 GpuSparse cuda_sparse(ctx); in operator ()() local
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc328 GpuSparse cuda_sparse(c); in operator ()() local
H A Ddense_to_csr_sparse_matrix_op.cc391 GpuSparse cuda_sparse(c); in operator ()() local