Searched refs:dev_workspace (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 391 auto dev_workspace = in PotrfImpl() local
396 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in PotrfImpl()
461 auto dev_workspace = in GetrfImpl() local
466 CUDAComplex(dev_workspace.mutable_data()), dev_pivots, dev_lapack_info)); in GetrfImpl()
521 auto dev_workspace = in GeqrfImpl() local
526 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in GeqrfImpl()
557 auto dev_workspace = in UnmqrImpl() local
563 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in UnmqrImpl()
601 auto dev_workspace = in UngqrImpl() local
606 CUDAComplex(dev_tau), CUDAComplex(dev_workspace.mutable_data()), in UngqrImpl()
[all …]
|