Searched refs:DN_SOLVER_FN (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_context.cc | 132 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method macro
138 TF_RETURN_IF_ERROR(CusolverStatusToStatus(DN_SOLVER_FN( \
150 return CusolverStatusToStatus(DN_SOLVER_FN(potrf, type_prefix)( \
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 301 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method macro
384 DN_SOLVER_FN(potrf, type_prefix), this, context_, \
417 DN_SOLVER_FN(getrf, type_prefix), this, context_, \
443 return GetrsImpl(DN_SOLVER_FN(getrs, type_prefix), context_, \
477 DN_SOLVER_FN(geqrf, type_prefix), this, context_, \
519 DN_SOLVER_FN(function_prefix##mqr, type_prefix), this, \
558 DN_SOLVER_FN(function_prefix##gqr, type_prefix), this, \
601 DN_SOLVER_FN(function_prefix##evd, type_prefix), this, \
639 DN_SOLVER_FN(gesvd, type_prefix), this, context_, \
682 DN_SOLVER_FN(gesvdjBatched, type_prefix), this, \
|