Home
last modified time | relevance | path

Searched refs:transA_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc533 transA_ = transpose_a ? (adjoint_a ? CUSPARSE_OPERATION_TRANSPOSE in CSRSparseSparseMatrixMatMul()
539 transA_ = transpose_a in CSRSparseSparseMatrixMatMul()
588 transA_, transB_, m, n, k, descrA_.descr(), nnzA, a.row_ptr.data(), in GetOutputStructure()
624 transA_, transB_, m, k, n, descrA_.descr(), nnzA, a.values.data(), in Compute()
652 gpusparseOperation_t transA_; member
Dmat_mul_op.cc816 : transA_(TransposeAndConjugateToGpuSparseOp(transpose_a, conjugate_a, in CSRSparseMatrixMatVec()
854 TF_RETURN_IF_ERROR(cuda_sparse.Csrmv(transA_, m, n, nnz, &alpha, descrA, in Compute()
864 const gpusparseOperation_t transA_; member in tensorflow::functor::CSRSparseMatrixMatVec