| /external/tensorflow/tensorflow/stream_executor/platform/default/ |
| D | dso_loader.cc | 43 port::StatusOr<void*> GetDsoHandle(const string& name, const string& version) { in GetDsoHandle() function
68 return GetDsoHandle("nvcuda", ""); in GetCudaDriverDsoHandle()
72 auto handle_or = GetDsoHandle("cuda", ""); in GetCudaDriverDsoHandle()
77 return GetDsoHandle("cuda", "1"); in GetCudaDriverDsoHandle()
81 return GetDsoHandle("cudart", GetCudaRtVersion()); in GetCudaRuntimeDsoHandle()
85 return GetDsoHandle("cublas", GetCublasVersion()); in GetCublasDsoHandle()
89 return GetDsoHandle("cublasLt", GetCublasVersion()); in GetCublasLtDsoHandle()
93 return GetDsoHandle("cufft", GetCufftVersion()); in GetCufftDsoHandle()
97 return GetDsoHandle("cusolver", GetCusolverVersion()); in GetCusolverDsoHandle()
101 return GetDsoHandle("cusparse", GetCusparseVersion()); in GetCusparseDsoHandle()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | curand_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
39 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cupti_stub.cc | 26 void* GetDsoHandle() { in GetDsoHandle() function
42 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cufft_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
39 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cublasLt_stub.cc | 24 void* GetDsoHandle() { in GetDsoHandle() function
41 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cusparse_stub.cc | 24 void* GetDsoHandle() { in GetDsoHandle() function
41 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cusolver_stub.cc | 24 void* GetDsoHandle() { in GetDsoHandle() function
41 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cublas_stub.cc | 29 void* GetDsoHandle() { in GetDsoHandle() function
45 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cudnn_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
39 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cuda_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
40 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cudart_stub.cc | 24 void* GetDsoHandle() { in GetDsoHandle() function
42 env->GetSymbolFromLibrary(GetDsoHandle(), symbol_name, &symbol).IgnoreError(); in LoadSymbol()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/stub/ |
| D | nvinfer_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
40 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | nvinfer_plugin_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
40 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | hipsparse_wrapper.h | 48 static void* GetDsoHandle() { \
56 Env::Default()->GetSymbolFromLibrary(GetDsoHandle(), kName, &f); \
|
| D | rocm_rng.cc | 81 static void* GetDsoHandle() { \
87 auto s = port::Env::Default()->GetSymbolFromLibrary(GetDsoHandle(), \
|
| D | rocm_fft.cc | 63 static void *GetDsoHandle() { \
69 auto s = port::Env::Default()->GetSymbolFromLibrary(GetDsoHandle(), \
|
| D | rocm_blas.cc | 74 static void *GetDsoHandle() { \
80 auto s = port::Env::Default()->GetSymbolFromLibrary(GetDsoHandle(), \
|
| D | rocm_dnn.cc | 225 static void* GetDsoHandle() { \
231 auto s = port::Env::Default()->GetSymbolFromLibrary(GetDsoHandle(), \
|
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.cc | 66 static void* GetDsoHandle() { \
73 GetDsoHandle(), kName, &f); \
|