Searched refs:CudaSolver (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 169 CudaSolver::CudaSolver(OpKernelContext* context) : context_(context) { in CudaSolver() function in tensorflow::CudaSolver
193 CudaSolver::~CudaSolver() { in ~CudaSolver()
200 void CudaSolver::CheckLapackInfoAndDeleteSolverAsync( in CheckLapackInfoAndDeleteSolverAsync()
201 std::unique_ptr<CudaSolver> solver, in CheckLapackInfoAndDeleteSolverAsync()
229 CudaSolver* solver, in CheckLapackInfoAndDeleteSolverAsync()
272 void CudaSolver::CheckLapackInfoAndDeleteSolverAsync( in CheckLapackInfoAndDeleteSolverAsync()
273 std::unique_ptr<CudaSolver> solver, in CheckLapackInfoAndDeleteSolverAsync()
294 Status CudaSolver::allocate_scoped_tensor(DataType type, in allocate_scoped_tensor()
304 Status CudaSolver::forward_input_or_allocate_scoped_tensor( in forward_input_or_allocate_scoped_tensor()
366 Status CudaSolver::Geam<Scalar>( \
[all …]
|
| D | cuda_solvers.h | 139 class CudaSolver {
142 explicit CudaSolver(OpKernelContext* context);
143 virtual ~CudaSolver();
155 std::unique_ptr<CudaSolver> solver,
164 std::unique_ptr<CudaSolver> solver,
370 TF_DISALLOW_COPY_AND_ASSIGN(CudaSolver);
465 ScratchSpace<Scalar> CudaSolver::GetScratchSpace(const TensorShape& shape,
474 ScratchSpace<Scalar> CudaSolver::GetScratchSpace(int64 size,
480 inline DeviceLapackInfo CudaSolver::GetDeviceLapackInfo(
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | svd_op_gpu.cu.cc | 100 std::unique_ptr<CudaSolver> solver) { in RunSVD()
270 std::unique_ptr<CudaSolver> solver) { in CheckResult()
282 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in CheckResult()
300 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in PerformSVD_MgeqN()
324 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in PerformSVD_MlessN()
|
| D | self_adjoint_eig_v2_op_gpu.cc | 85 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
156 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | determinant_op.cc | 168 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
262 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
313 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
409 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | cholesky_op.cc | 124 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
205 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | tridiagonal_solve_op_gpu.cu.cc | 170 std::unique_ptr<CudaSolver> cublas_solver(new CudaSolver(context)); in ComputeMatrix()
187 const std::unique_ptr<CudaSolver>& cublas_solver, in TransposeWithGeam()
348 std::unique_ptr<CudaSolver> cublas_solver(new CudaSolver(context)); in TransposeLhsForGtsvBatched()
|
| D | matrix_inverse_op.cc | 128 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
261 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | qr_op_impl.h | 174 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
288 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | matrix_solve_op.cc | 171 std::unique_ptr<CudaSolver> solver(new CudaSolver(context)); in ComputeAsync()
352 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | lu_op_gpu.cu.cc | 111 auto solver = absl::make_unique<CudaSolver>(context); in ComputeAsync()
254 CudaSolver::CheckLapackInfoAndDeleteSolverAsync(std::move(solver), dev_info, in ComputeAsync()
|
| D | matrix_triangular_solve_op_impl.h | 320 auto solver = absl::make_unique<CudaSolver>(context);
|