Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc113 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule() local
124 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) { in UnloadGpuBinary()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc295 bool GpuExecutor::UnloadGpuBinary(const void* gpu_binary) { in UnloadGpuBinary()
361 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_jitrt.cc374 auto gpu_binary = gpu_module->getAttrOfType<mlir::StringAttr>("binary"); in matchAndRewrite() local