Searched defs:cuda_solver (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 359 CudaSolver* cuda_solver, in PotrfImpl()
392 CudaSolver* cuda_solver, in GetrfImpl()
452 CudaSolver* cuda_solver, in GeqrfImpl()
485 CudaSolver* cuda_solver, in UnmqrImpl()
531 CudaSolver* cuda_solver, in UngqrImpl()
570 CudaSolver* cuda_solver, in HeevdImpl()
613 BufSizeFnT bufsize, SolverFnT solver, CudaSolver* cuda_solver, in GesvdImpl()
648 CudaSolver* cuda_solver, in GesvdjBatchedImpl()
699 static inline Status GetrfBatchedImpl(SolverFnT solver, CudaSolver* cuda_solver, in GetrfBatchedImpl()
736 SolverFnT solver, CudaSolver* cuda_solver, OpKernelContext* context, in GetrsBatchedImpl()
[all …]
|