Searched refs:DN_SOLVER_FN (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 322 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method macro
405 DN_SOLVER_FN(potrf, type_prefix), this, context_, \
439 return PotrfBatchedImpl(DN_SOLVER_FN(potrfBatched, type_prefix), this, \
475 DN_SOLVER_FN(getrf, type_prefix), this, context_, \
501 return GetrsImpl(DN_SOLVER_FN(getrs, type_prefix), context_, \
535 DN_SOLVER_FN(geqrf, type_prefix), this, context_, \
577 DN_SOLVER_FN(function_prefix##mqr, type_prefix), this, \
616 DN_SOLVER_FN(function_prefix##gqr, type_prefix), this, \
659 DN_SOLVER_FN(function_prefix##evd, type_prefix), this, \
697 DN_SOLVER_FN(gesvd, type_prefix), this, context_, \
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_context.cc | 134 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method macro
176 return CusolverStatusToStatus(DN_SOLVER_FN(potrf, type_prefix)( \
|