Searched refs:GpuModuleHandle (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 273 bool LoadModuleFromCuBin(const char* cubin, GpuModuleHandle* module)
278 bool LoadModuleFromPtx(const char* ptx, GpuModuleHandle* module)
282 bool LoadModuleFromHsaco(const char* hsaco, GpuModuleHandle* module)
295 std::map<string, GpuModuleHandle> disk_modules_ GUARDED_BY(disk_modules_mu_);
300 std::map<const char*, GpuModuleHandle> in_memory_modules_
307 std::unordered_map<const void*, std::pair<GpuModuleHandle, uint64>>
|
| D | gpu_driver.h | 215 GpuModuleHandle* module);
221 GpuModuleHandle* module);
227 GpuModuleHandle* module);
232 static bool GetModuleFunction(GpuContext* context, GpuModuleHandle module,
240 static bool GetModuleSymbol(GpuContext* context, GpuModuleHandle module,
248 static void UnloadModule(GpuContext* context, GpuModuleHandle module);
|
| D | gpu_types.h | 55 using GpuModuleHandle = hipModule_t; variable
73 using GpuModuleHandle = CUmodule;
|