Home
last modified time | relevance | path

Searched refs:module_it (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc123 auto module_it = gpu_binary_to_module_.find(gpu_binary); in UnloadGpuBinary() local
124 if (gpu_binary_to_module_.end() == module_it) { in UnloadGpuBinary()
128 auto& module = module_it->second.first; in UnloadGpuBinary()
129 auto& refcount = module_it->second.second; in UnloadGpuBinary()
134 gpu_binary_to_module_.erase(module_it); in UnloadGpuBinary()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc316 auto module_it = gpu_binary_to_module_.find(gpu_binary); in UnloadGpuBinary() local
317 if (gpu_binary_to_module_.end() == module_it) { in UnloadGpuBinary()
321 auto& module = module_it->second.first; in UnloadGpuBinary()
322 auto& refcount = module_it->second.second; in UnloadGpuBinary()
327 gpu_binary_to_module_.erase(module_it); in UnloadGpuBinary()