Searched defs: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()
414 SolverFnT solver, CudaSolver* cuda_solver, OpKernelContext* context, in PotrfBatchedImpl()
450 CudaSolver* cuda_solver, in GetrfImpl()
510 CudaSolver* cuda_solver, in GeqrfImpl()
543 CudaSolver* cuda_solver, in UnmqrImpl()
589 CudaSolver* cuda_solver, in UngqrImpl()
628 CudaSolver* cuda_solver, in HeevdImpl()
671 BufSizeFnT bufsize, SolverFnT solver, CudaSolver* cuda_solver, in GesvdImpl()
706 CudaSolver* cuda_solver, in GesvdjBatchedImpl()
757 static inline Status GetrfBatchedImpl(SolverFnT solver, CudaSolver* cuda_solver, in GetrfBatchedImpl()
[all …]
|