Searched refs:transA_ (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | sparse_mat_mul_op.cc | 533 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
|
D | mat_mul_op.cc | 816 : 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
|