Searched refs:BLAS_SOLVER_FN (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 328 #define BLAS_SOLVER_FN(method, type_prefix) cublas##type_prefix##method macro
372 return GeamImpl(BLAS_SOLVER_FN(geam, type_prefix), cublas_handle_, transa, \
785 return GetrfBatchedImpl(BLAS_SOLVER_FN(getrfBatched, type_prefix), this, \
831 BLAS_SOLVER_FN(getrsBatched, type_prefix)), \
875 BLAS_SOLVER_FN(getriBatched, type_prefix)), \
916 BLAS_SOLVER_FN(matinvBatched, type_prefix)), \
947 return TrsmImpl(BLAS_SOLVER_FN(trsm, type_prefix), cublas_handle_, side, \
971 return TrsvImpl(BLAS_SOLVER_FN(trsv, type_prefix), cublas_handle_, uplo, \
1017 BLAS_SOLVER_FN(trsmBatched, type_prefix)), \
|
| D | rocm_solvers.cc | 180 #define BLAS_SOLVER_FN(method, type_prefix) \ macro
236 return TrsmImpl(gpu_executor, BLAS_SOLVER_FN(trsm, type_prefix), \
|