Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc148 auto gpu_binary_it = kernel_to_gpu_binary_.find(kernel); in UnloadKernel() local
149 if (kernel_to_gpu_binary_.end() == gpu_binary_it) { in UnloadKernel()
155 << " has loaded GPU code " << gpu_binary_it->second; in UnloadKernel()
156 UnloadGpuBinary(gpu_binary_it->second); in UnloadKernel()
157 kernel_to_gpu_binary_.erase(gpu_binary_it); in UnloadKernel()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc336 auto gpu_binary_it = kernel_to_gpu_binary_.find(kernel); in UnloadKernel() local
337 if (kernel_to_gpu_binary_.end() == gpu_binary_it) { in UnloadKernel()
343 << " has loaded GPU code " << gpu_binary_it->second; in UnloadKernel()
344 UnloadGpuBinary(gpu_binary_it->second); in UnloadKernel()
345 kernel_to_gpu_binary_.erase(gpu_binary_it); in UnloadKernel()