Home
last modified time | relevance | path

Searched refs:GetModuleSymbol (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc742 if (GpuDriver::GetModuleSymbol( in GetSymbol()
750 if (GpuDriver::GetModuleSymbol( in GetSymbol()
H A Drocm_driver.cc537 /* static */ bool GpuDriver::GetModuleSymbol(GpuContext* context, in GetModuleSymbol() function in stream_executor::gpu::GpuDriver
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dgpu_driver.h301 static bool GetModuleSymbol(GpuContext* context, GpuModuleHandle module,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc874 return GpuDriver::GetModuleSymbol(context_, module, symbol_name.c_str(), in GetSymbol()
H A Dcuda_driver.cc634 /* static */ bool GpuDriver::GetModuleSymbol(GpuContext* context, in GetModuleSymbol() function in stream_executor::gpu::GpuDriver