Home
last modified time | relevance | path

Searched refs:dev_lapack_info (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcuda_solvers.cc192 for (const auto& dev_lapack_info : dev_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local
194 auto host_copy = dev_lapack_info.CopyToHost(&success); in CheckLapackInfoAndDeleteSolverAsync()
199 dev_lapack_info.debug_info())); in CheckLapackInfoAndDeleteSolverAsync()
253 const std::vector<DeviceLapackInfo>& dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync() argument
266 CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync()
363 int* dev_lapack_info) { in PotrfImpl() argument
375 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in PotrfImpl()
382 int lda, int* dev_lapack_info) { \
385 cusolver_dn_handle_, uplo, n, A, lda, dev_lapack_info); \
396 int* dev_lapack_info) { in GetrfImpl() argument
[all …]
Dcuda_solvers.h151 const std::vector<DeviceLapackInfo>& dev_lapack_info,
160 const std::vector<DeviceLapackInfo>& dev_lapack_info,
213 int* dev_lapack_info) TF_MUST_USE_RESULT;
220 int* dev_lapack_info) TF_MUST_USE_RESULT;
227 int* dev_lapack_info) const TF_MUST_USE_RESULT;
234 int* dev_pivots, DeviceLapackInfo* dev_lapack_info,
256 DeviceLapackInfo* dev_lapack_info,
265 int ldainv, DeviceLapackInfo* dev_lapack_info,
274 int* dev_lapack_info) TF_MUST_USE_RESULT;
287 Scalar* dev_c, int ldc, int* dev_lapack_info) TF_MUST_USE_RESULT;
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.h51 int lda, se::DeviceMemory<int> dev_lapack_info,
54 int lda, se::DeviceMemory<int> dev_lapack_info,
58 se::DeviceMemory<int> dev_lapack_info,
62 se::DeviceMemory<int> dev_lapack_info,