Searched refs: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
304 cusolverDn##type_prefix##method##_bufferSize
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
351 return GeamImpl(BLAS_SOLVER_FN(geam, type_prefix), cublas_handle_, transa, \
379 #define POTRF_INSTANCE(Scalar, type_prefix) \ argument
383 return PotrfImpl(DN_BUFSIZE_FN(potrf, type_prefix), \
[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
139 potrf_bufferSize, type_prefix)(handle(), CUDABlasUpperLower(uplo), n, \
146 #define POTRF_INSTANCE(T, type_prefix) \ argument
150 return CusolverStatusToStatus(DN_SOLVER_FN(potrf, type_prefix)( \
|