Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcuda_solvers.h76 class HostLapackInfo;
152 std::function<void(const Status&, const std::vector<HostLapackInfo>&)>
391 class HostLapackInfo : public ScratchSpace<int> {
393 HostLapackInfo(OpKernelContext* context, int64 size, const string& debug_info)
406 HostLapackInfo CopyToHost(bool* success) const {
408 HostLapackInfo copy(context(), size(), debug_info());
Dcuda_solvers.cc182 std::function<void(const Status&, const std::vector<HostLapackInfo>&)> in CheckLapackInfoAndDeleteSolverAsync()
185 std::vector<HostLapackInfo> host_lapack_infos; in CheckLapackInfoAndDeleteSolverAsync()
209 std::function<void(const Status&, const std::vector<HostLapackInfo>&)> in CheckLapackInfoAndDeleteSolverAsync()
211 std::vector<HostLapackInfo> host_lapack_infos) { in CheckLapackInfoAndDeleteSolverAsync()
258 const std::vector<HostLapackInfo>& /* unused */) { in CheckLapackInfoAndDeleteSolverAsync() argument
Dcholesky_op.cc160 const std::vector<HostLapackInfo>& /* unused */) { in ComputeAsync() argument
Ddeterminant_op.cc245 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
392 const std::vector<HostLapackInfo>& host_infos) { in ComputeAsync()
Dlu_op_gpu.cu.cc237 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.cc222 const std::vector<HostLapackInfo>& /* unused */) { in CheckResult() argument