Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc117 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule() local
119 return UnloadGpuBinary(gpu_binary); in UnloadModule()
122 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) { in UnloadGpuBinary() argument
123 auto module_it = gpu_binary_to_module_.find(gpu_binary); in UnloadGpuBinary()
125 VLOG(3) << "No loaded HSACO module for " << gpu_binary; in UnloadGpuBinary()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc315 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) { in UnloadGpuBinary() argument
316 auto module_it = gpu_binary_to_module_.find(gpu_binary); in UnloadGpuBinary()
318 VLOG(3) << "No loaded CUDA module for " << gpu_binary; in UnloadGpuBinary()
381 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule() local
383 return UnloadGpuBinary(gpu_binary); in UnloadModule()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h283 bool UnloadGpuBinary(const void* gpu_binary)