Home
last modified time | relevance | path

Searched refs:CreateCSRSparseMatrix (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dconj_op.cc55 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator ()()
77 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( \
H A Dsoftmax_op.cc70 CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
177 CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc115 ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
281 CSRSparseMatrix::CreateCSRSparseMatrix( in ComputeAsync()
H A Ddense_to_csr_sparse_matrix_op.cc120 OP_REQUIRES_OK(ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
333 CSRSparseMatrix::CreateCSRSparseMatrix( in ComputeAsync()
H A Dzeros_op.h73 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator()
H A Dsparse_matrix.h145 static Status CreateCSRSparseMatrix(DataType dtype, in CreateCSRSparseMatrix() function
393 return CreateCSRSparseMatrix(from.dtype(), in DeviceCopy()
H A Dmul_op.cc134 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
H A Dsparse_mat_mul_op.cc248 OP_REQUIRES_OK(ctx, CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
483 CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
H A Dsparse_cholesky_op.cc217 CSRSparseMatrix::CreateCSRSparseMatrix( in Compute()
H A Dtranspose_op.cc179 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator ()()
H A Dadd_op.cc158 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in operator ()()
H A Dkernels_gpu.cu.cc293 TF_RETURN_IF_ERROR(CSRSparseMatrix::CreateCSRSparseMatrix( in CSRSparseMatrixBatchMulVecImpl()