Searched refs:cuda_solver (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 380 CudaSolver* cuda_solver, in PotrfImpl() argument
392 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false); in PotrfImpl()
414 SolverFnT solver, CudaSolver* cuda_solver, OpKernelContext* context, in PotrfBatchedImpl() argument
421 cuda_solver->GetScratchSpace<uint8>(sizeof(CudaScalar*) * batch_size, "", in PotrfBatchedImpl()
450 CudaSolver* cuda_solver, in GetrfImpl() argument
462 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false); in GetrfImpl()
510 CudaSolver* cuda_solver, in GeqrfImpl() argument
522 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false); in GeqrfImpl()
543 CudaSolver* cuda_solver, in UnmqrImpl() argument
558 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false); in UnmqrImpl()
[all …]
|