Home
last modified time | relevance | path

Searched refs:in_memory_modules_mu_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h272 TF_EXCLUSIVE_LOCKS_REQUIRED(in_memory_modules_mu_);
277 TF_EXCLUSIVE_LOCKS_REQUIRED(in_memory_modules_mu_);
281 TF_EXCLUSIVE_LOCKS_REQUIRED(in_memory_modules_mu_);
284 TF_EXCLUSIVE_LOCKS_REQUIRED(in_memory_modules_mu_);
297 absl::Mutex in_memory_modules_mu_; variable
300 TF_GUARDED_BY(in_memory_modules_mu_);
304 TF_GUARDED_BY(in_memory_modules_mu_);
307 gpu_binary_to_module_ TF_GUARDED_BY(in_memory_modules_mu_);
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc272 absl::MutexLock lock{&in_memory_modules_mu_}; in GetKernel()
292 absl::MutexLock lock{&in_memory_modules_mu_}; in GetKernel()
335 absl::MutexLock lock{&in_memory_modules_mu_}; in UnloadKernel()
354 absl::MutexLock lock{&in_memory_modules_mu_}; in LoadModule()
370 absl::MutexLock lock{&in_memory_modules_mu_}; in LoadModule()
382 absl::MutexLock lock{&in_memory_modules_mu_}; in UnloadModule()
830 absl::MutexLock lock{&in_memory_modules_mu_}; in GetSymbol()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc118 absl::MutexLock lock{&in_memory_modules_mu_}; in UnloadModule()
147 absl::MutexLock lock{&in_memory_modules_mu_}; in UnloadKernel()
257 absl::MutexLock lock{&in_memory_modules_mu_}; in GetKernel()
374 absl::MutexLock lock{&in_memory_modules_mu_}; in LoadModule()
730 absl::MutexLock lock{&in_memory_modules_mu_}; in GetSymbol()