Searched defs:alpha_host (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | rocm_sparse.cc | 183 int k, int nnz, const Scalar* alpha_host, const hipsparseMatDescr_t descrA, in CsrmmImpl()
216 int nnz, const Scalar* alpha_host, in CsrmvImpl()
|
| D | cuda_sparse.cc | 443 int nnz, const Scalar* alpha_host, const cusparseMatDescr_t descrA, in CsrmmImpl()
482 cusparseOperation_t transA, int m, int n, int nnz, const Scalar* alpha_host, in CsrmvImpl()
|