Home
last modified time | relevance | path

Searched refs:HostLapackInfo (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.h81 class HostLapackInfo;
157 std::function<void(const Status&, const std::vector<HostLapackInfo>&)>
434 class HostLapackInfo : public ScratchSpace<int> {
436 HostLapackInfo(OpKernelContext* context, int64 size,
450 HostLapackInfo CopyToHost(bool* success) const {
452 HostLapackInfo copy(context(), size(), debug_info());
Dcuda_solvers.cc203 std::function<void(const Status&, const std::vector<HostLapackInfo>&)> in CheckLapackInfoAndDeleteSolverAsync()
206 std::vector<HostLapackInfo> host_lapack_infos; in CheckLapackInfoAndDeleteSolverAsync()
230 std::function<void(const Status&, const std::vector<HostLapackInfo>&)> in CheckLapackInfoAndDeleteSolverAsync()
232 std::vector<HostLapackInfo> host_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync()
279 const std::vector<HostLapackInfo>& /* unused */) { in CheckLapackInfoAndDeleteSolverAsync() argument
/external/tensorflow/tensorflow/core/kernels/linalg/
Ddeterminant_op.cc245 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
392 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
Dcholesky_op.cc200 const std::vector<HostLapackInfo>& /* unused */) { in ComputeAsync() argument
Dlu_op_gpu.cu.cc239 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
Dmatrix_inverse_op.cc245 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
Dmatrix_solve_op.cc339 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
Dsvd_op_gpu.cu.cc273 const std::vector<HostLapackInfo>& /* unused */) { in CheckResult() argument