Searched defs:type_prefix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 301 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method argument
302 #define DN_SOLVER_NAME(method, type_prefix) "cusolverDn" #type_prefix #method argument
303 #define DN_BUFSIZE_FN(method, type_prefix) \ argument
307 #define BLAS_SOLVER_FN(method, type_prefix) cublas##type_prefix##method argument
308 #define BLAS_SOLVER_NAME(method, type_prefix) "cublas" #type_prefix #method argument
343 #define GEAM_INSTANCE(Scalar, type_prefix) \ argument
379 #define POTRF_INSTANCE(Scalar, type_prefix) \ argument
412 #define GETRF_INSTANCE(Scalar, type_prefix) \ argument
438 #define GETRS_INSTANCE(Scalar, type_prefix) \ argument
472 #define GEQRF_INSTANCE(Scalar, type_prefix) \ argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_context.cc | 132 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method argument
134 #define POTRF_BUFFER_SIZE_INSTANCE(T, type_prefix) \ argument
146 #define POTRF_INSTANCE(T, type_prefix) \ argument
|