Lines Matching defs:dev_info
203 std::vector<DeviceLapackInfo> dev_info;
212 dev_info.push_back(
217 &dev_info.back(), batch_size),
222 dev_info.push_back(solver->GetDeviceLapackInfo(batch_size, "getrf"));
227 &pivots_mat(batch, 0), &dev_info.back()(batch)),
240 pivots_mat.data(), output_reshaped, dev_info.back().mutable_data());
262 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info,
347 std::vector<DeviceLapackInfo> dev_info;
356 dev_info.push_back(
361 &dev_info.back(), batch_size),
366 dev_info.push_back(solver->GetDeviceLapackInfo(batch_size, "getrf"));
371 &pivots_mat(batch, 0), &dev_info.back()(batch)),
409 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info,
