Searched refs:transA_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_mat_mul_op.cc | 568 transA_ = transpose_a in CSRSparseSparseMatrixMatMul()
657 transA_, transB_, m, n, k, descrA_.descr(), nnzA, a.row_ptr.data(), in CSRSparseSparseMatrixMatMul()
700 transA_, transB_, m, k, n, descrA_.descr(), nnzA, a.values.data(), in CSRSparseSparseMatrixMatMul()
735 gpusparseOperation_t transA_; in CSRSparseSparseMatrixMatMul() member
|
| D | mat_mul_op.cc | 915 : transA_(TransposeAndConjugateToGpuSparseOp(transpose_a, conjugate_a, in CSRSparseMatrixMatVec()
955 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, in Compute()
959 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, descrA, in Compute()
970 const gpusparseOperation_t transA_; member in tensorflow::functor::CSRSparseMatrixMatVec
|