Searched refs:host_lapack_info (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 235 for (const auto& host_lapack_info : host_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local
236 for (int i = 0; i < host_lapack_info.size() && status.ok(); ++i) { in CheckLapackInfoAndDeleteSolverAsync()
237 const int info_value = host_lapack_info(i); in CheckLapackInfoAndDeleteSolverAsync()
242 host_lapack_info.debug_info()); in CheckLapackInfoAndDeleteSolverAsync()
797 const Scalar* const host_b_dev_ptrs[], int ldb, int* host_lapack_info, in GetrsBatchedImpl() argument
819 ldb, host_lapack_info, batch_size)); in GetrsBatchedImpl()
828 const Scalar* const host_b_dev_ptrs[], int ldb, int* host_lapack_info, \
834 ldb, host_lapack_info, batch_size); \
|
| D | cuda_solvers.h | 264 int ldb, int* host_lapack_info,
|