Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc150 auto gpu_binary_it = kernel_to_gpu_binary_.find(kernel); in UnloadKernel() local
151 if (kernel_to_gpu_binary_.end() == gpu_binary_it) { in UnloadKernel()
157 << " has loaded GPU code " << gpu_binary_it->second; in UnloadKernel()
158 UnloadGpuBinary(gpu_binary_it->second); in UnloadKernel()
159 kernel_to_gpu_binary_.erase(gpu_binary_it); in UnloadKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc316 auto gpu_binary_it = kernel_to_gpu_binary_.find(kernel); in UnloadKernel() local
317 if (kernel_to_gpu_binary_.end() == gpu_binary_it) { in UnloadKernel()
323 << " has loaded GPU code " << gpu_binary_it->second; in UnloadKernel()
324 UnloadGpuBinary(gpu_binary_it->second); in UnloadKernel()
325 kernel_to_gpu_binary_.erase(gpu_binary_it); in UnloadKernel()