Searched defs:dev_lapack_info (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 192 for (const auto& dev_lapack_info : dev_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local
253 const std::vector<DeviceLapackInfo>& dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync()
363 int* dev_lapack_info) { in PotrfImpl()
396 int* dev_lapack_info) { in GetrfImpl()
429 Scalar* B, int ldb, int* dev_lapack_info) { in GetrsImpl()
456 int* dev_lapack_info) { in GeqrfImpl()
491 int ldc, int* dev_lapack_info) { in UnmqrImpl()
535 const Scalar* dev_tau, int* dev_lapack_info) { in UngqrImpl()
576 int* dev_lapack_info) { in HeevdImpl()
616 Scalar* S, Scalar* U, int ldu, Scalar* VT, int ldvt, int* dev_lapack_info) { in GesvdImpl()
[all …]
|