Searched refs:dev_pivots (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 395 int n, Scalar* A, int lda, int* dev_pivots, in GetrfImpl() argument
408 CUDAComplex(dev_workspace.mutable_data()), dev_pivots, dev_lapack_info)); in GetrfImpl()
415 int* dev_pivots, int* dev_lapack_info) { \
418 cusolver_dn_handle_, m, n, A, lda, dev_pivots, \
703 int lda, int* dev_pivots, in GetrfBatchedImpl() argument
718 dev_pivots, dev_lapack_info->mutable_data(), batch_size)); in GetrfBatchedImpl()
725 int n, const Scalar* const host_a_dev_ptrs[], int lda, int* dev_pivots, \
729 dev_pivots, dev_lapack_info, batch_size); \
738 const Scalar* const host_a_dev_ptrs[], int lda, const int* dev_pivots, in GetrsBatchedImpl() argument
760 dev_pivots, reinterpret_cast<CudaScalar**>(dev_b_dev_ptrs.mutable_data()), in GetrsBatchedImpl()
[all …]
|
| D | cuda_solvers.h | 219 Status Getrf(int m, int n, Scalar* dev_A, int lda, int* dev_pivots,
234 int* dev_pivots, DeviceLapackInfo* dev_lapack_info,
254 const int* dev_pivots,
|