Searched refs:cublasMath_t (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cublas_stub.cc | 61 typedef enum {} cublasMath_t; typedef
|
| D | cuda_blas.h | 92 cublasMath_t math_type, Args... args);
|
| D | cuda_blas.cc | 188 bool Init(cublasMath_t new_mode) { in Init()
217 cublasMath_t old_mode_; // Prior cuBLAS math mode, to be restored.
534 cublasMath_t math_type, Args... args) { in DoBlasInternalImpl()
1775 cublasMath_t math_type = CUBLAS_TENSOR_OP_MATH; in DoBlasInternalImpl()
1777 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl()
1829 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl()
1831 cublasMath_t math_type = CUBLAS_TF32_TENSOR_OP_MATH; in DoBlasInternalImpl()
2085 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl()
2437 cublasMath_t math_type; in DoBlasInternalImpl()
2613 cublasMath_t math_type = CUBLAS_TENSOR_OP_MATH; in DoBlasInternalImpl()
[all …]
|
| D | cublas_9_0.inc | 89 cublasMath_t *mode) {
91 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t *);
98 cublasMath_t mode) {
99 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t);
|
| D | cublas_11_0.inc | 96 cublasMath_t *mode) {
98 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t *);
105 cublasMath_t mode) {
106 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t);
|
| D | cublas_10_1.inc | 96 cublasMath_t *mode) {
98 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t *);
105 cublasMath_t mode) {
106 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t);
|
| D | cublas_10_0.inc | 89 cublasMath_t *mode) {
91 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t *);
98 cublasMath_t mode) {
99 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t);
|
| D | cublas_10_2.inc | 96 cublasMath_t *mode) {
98 cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t *);
105 cublasMath_t mode) {
106 using FuncPtr = cublasStatus_t(CUBLASWINAPI *)(cublasHandle_t, cublasMath_t);
|