Searched refs:GetDsoHandle (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/platform/default/ |
| D | dso_loader.cc | 36 port::StatusOr<void*> GetDsoHandle(const string& name, const string& version) { in GetDsoHandle() function
61 return GetDsoHandle("nvcuda", ""); in GetCudaDriverDsoHandle()
65 auto handle_or = GetDsoHandle("cuda", ""); in GetCudaDriverDsoHandle()
70 return GetDsoHandle("cuda", "1"); in GetCudaDriverDsoHandle()
74 return GetDsoHandle("cudart", GetCudaVersion()); in GetCudaRuntimeDsoHandle()
78 return GetDsoHandle("cublas", GetCudaVersion()); in GetCublasDsoHandle()
82 return GetDsoHandle("cufft", GetCudaVersion()); in GetCufftDsoHandle()
86 return GetDsoHandle("curand", GetCudaVersion()); in GetCurandDsoHandle()
93 return GetDsoHandle("cupti", ""); in GetCuptiDsoHandle()
95 return GetDsoHandle("cupti", GetCudaVersion()); in GetCuptiDsoHandle()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cufft_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
39 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | curand_stub.cc | 23 void* GetDsoHandle() { in GetDsoHandle() function
39 if (auto handle = GetDsoHandle()) { in LoadSymbol()
|
| D | cublas_stub.cc | 25 void* GetDsoHandle() { in GetDsoHandle() function
41 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
38 env->GetSymbolFromLibrary(GetDsoHandle(), symbol_name, &symbol).IgnoreError(); in LoadSymbol()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_rng.cc | 82 static void* GetDsoHandle() { \
88 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 | 134 static void* GetDsoHandle() { \
140 auto s = port::Env::Default()->GetSymbolFromLibrary(GetDsoHandle(), \
|