Searched defs:alpha_host (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_sparse.cc | 184 int k, int nnz, const Scalar* alpha_host, const hipsparseMatDescr_t descrA, in CsrmmImpl()
217 int nnz, const Scalar* alpha_host, in CsrmvImpl()
|
| D | cuda_sparse.cc | 395 int nnz, const Scalar* alpha_host, const cusparseMatDescr_t descrA, in CsrmmImpl()
473 cusparseOperation_t transA, int m, int n, int nnz, const Scalar* alpha_host, in CsrmvImpl()
515 int nnz, const Scalar* alpha_host, in CsrmvExImpl()
554 const Scalar* alpha_host, in SpMVImpl()
|