Searched defs:alpha_host (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A 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()
|
| H A D | rocm_sparse.cc | 192 int k, int nnz, const Scalar* alpha_host, const hipsparseMatDescr_t descrA, in CsrmmImpl()
263 int nnz, const Scalar* alpha_host, in CsrmvImpl()
|