Searched refs:host_a_dev_ptrs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 416 const Scalar* const host_a_dev_ptrs[], int lda, in PotrfBatchedImpl() argument
424 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in PotrfBatchedImpl()
437 cublasFillMode_t uplo, int n, const Scalar* const host_a_dev_ptrs[], \
441 host_a_dev_ptrs, lda, dev_lapack_info, \
760 const Scalar* const host_a_dev_ptrs[], in GetrfBatchedImpl() argument
770 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in GetrfBatchedImpl()
783 int n, const Scalar* const host_a_dev_ptrs[], int lda, int* dev_pivots, \
786 context_, cublas_handle_, n, host_a_dev_ptrs, lda, \
796 const Scalar* const host_a_dev_ptrs[], int lda, const int* dev_pivots, in GetrsBatchedImpl() argument
808 host_a_dev_ptrs /* source */, dev_a_dev_ptrs.bytes())) { in GetrsBatchedImpl()
[all …]
|
| D | cuda_solvers.h | 229 const Scalar* const host_a_dev_ptrs[], int lda,
252 Status GetrfBatched(int n, const Scalar* const host_a_dev_ptrs[], int lda,
272 Status GetriBatched(int n, const Scalar* const host_a_dev_ptrs[], int lda,
282 Status MatInvBatched(int n, const Scalar* const host_a_dev_ptrs[], int lda,
|