Searched defs:dev_lapack_info (Results 1 – 2 of 2) sorted by relevance
212 for (const auto& dev_lapack_info : dev_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local273 const std::vector<DeviceLapackInfo>& dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync()382 int* dev_lapack_info) { in PotrfImpl()415 DeviceLapackInfo* dev_lapack_info, int batch_size) { in PotrfBatchedImpl()451 int* dev_lapack_info) { in GetrfImpl()484 Scalar* B, int ldb, int* dev_lapack_info) { in GetrsImpl()510 int* dev_lapack_info) { in GeqrfImpl()544 int ldc, int* dev_lapack_info) { in UnmqrImpl()587 const Scalar* dev_tau, int* dev_lapack_info) { in UngqrImpl()627 int* dev_lapack_info) { in HeevdImpl()[all …]
167 for (const auto& dev_lapack_info : dev_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local228 const std::vector<DeviceLapackInfo>& dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync()774 int ldainv, DeviceLapackInfo* dev_lapack_info, in MatInvBatchedImpl()