Home
last modified time | relevance | path

Searched defs:COOSparseMatrixToCSRSparseMatrix (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc324 struct COOSparseMatrixToCSRSparseMatrix<GPUDevice> { struct
325 Status operator()(OpKernelContext* c, const int rows, const int cols, in operator ()()
H A Ddense_to_csr_sparse_matrix_op.cc387 struct COOSparseMatrixToCSRSparseMatrix<GPUDevice> { struct
388 Status operator()(OpKernelContext* c, const int rows, const int cols, in operator ()()
H A Dkernels.h101 struct COOSparseMatrixToCSRSparseMatrix { struct