Searched refs:host_lapack_info (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 214 for (const auto& host_lapack_info : host_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local
215 for (int i = 0; i < host_lapack_info.size() && status.ok(); ++i) { in CheckLapackInfoAndDeleteSolverAsync()
216 const int info_value = host_lapack_info(i); in CheckLapackInfoAndDeleteSolverAsync()
221 host_lapack_info.debug_info()); in CheckLapackInfoAndDeleteSolverAsync()
739 const Scalar* const host_b_dev_ptrs[], int ldb, int* host_lapack_info, in GetrsBatchedImpl() argument
761 ldb, host_lapack_info, batch_size)); in GetrsBatchedImpl()
770 const Scalar* const host_b_dev_ptrs[], int ldb, int* host_lapack_info, \
776 ldb, host_lapack_info, batch_size); \
|
| D | cuda_solvers.h | 245 int ldb, int* host_lapack_info,
|