Searched defs:cuda_solver (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_solvers.cc | 379 GpuSolver* cuda_solver, OpKernelContext* context, in PotrfImpl()
412 SolverFnT solver, GpuSolver* cuda_solver, OpKernelContext* context, in PotrfBatchedImpl()
448 GpuSolver* cuda_solver, OpKernelContext* context, in GetrfImpl()
507 GpuSolver* cuda_solver, OpKernelContext* context, in GeqrfImpl()
539 GpuSolver* cuda_solver, OpKernelContext* context, in UnmqrImpl()
584 GpuSolver* cuda_solver, OpKernelContext* context, in UngqrImpl()
622 GpuSolver* cuda_solver, OpKernelContext* context, in HeevdImpl()
676 GpuSolver* cuda_solver, OpKernelContext* context, in GesvdImpl()
713 GpuSolver* cuda_solver, in GesvdjBatchedImpl()
764 static inline Status GetrfBatchedImpl(SolverFnT solver, GpuSolver* cuda_solver, in GetrfBatchedImpl()
[all …]
|