Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc213 for (const auto& dev_lapack_info : dev_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync() local
215 auto host_copy = dev_lapack_info.CopyToHost(&success); in CheckLapackInfoAndDeleteSolverAsync()
220 dev_lapack_info.debug_info())); in CheckLapackInfoAndDeleteSolverAsync()
274 const std::vector<DeviceLapackInfo>& dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync() argument
287 CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_lapack_info, in CheckLapackInfoAndDeleteSolverAsync()
384 int* dev_lapack_info) { in PotrfImpl() argument
396 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in PotrfImpl()
403 int lda, int* dev_lapack_info) { \
406 cusolver_dn_handle_, uplo, n, A, lda, dev_lapack_info); \
417 DeviceLapackInfo* dev_lapack_info, int batch_size) { in PotrfBatchedImpl() argument
[all …]
Dcuda_solvers.h156 const std::vector<DeviceLapackInfo>& dev_lapack_info,
165 const std::vector<DeviceLapackInfo>& dev_lapack_info,
220 int* dev_lapack_info) TF_MUST_USE_RESULT;
230 DeviceLapackInfo* dev_lapack_info,
239 int* dev_lapack_info) TF_MUST_USE_RESULT;
246 int* dev_lapack_info) const TF_MUST_USE_RESULT;
253 int* dev_pivots, DeviceLapackInfo* dev_lapack_info,
275 DeviceLapackInfo* dev_lapack_info,
284 int ldainv, DeviceLapackInfo* dev_lapack_info,
293 int* dev_lapack_info) TF_MUST_USE_RESULT;
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.h59 int lda, se::DeviceMemory<int> dev_lapack_info,
104 int lda, se::DeviceMemory<int> dev_lapack_info,