Searched refs:GpuModuleHandle (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 271 port::Status LoadModuleFromCuBin(const char* cubin, GpuModuleHandle* module)
276 port::Status LoadModuleFromPtx(const char* ptx, GpuModuleHandle* module)
280 port::Status LoadModuleFromHsaco(const char* hsaco, GpuModuleHandle* module)
293 std::map<std::string, GpuModuleHandle> disk_modules_
299 std::map<const char*, GpuModuleHandle> in_memory_modules_
306 std::unordered_map<const void*, std::pair<GpuModuleHandle, uint64>>
|
| D | gpu_driver.h | 271 GpuModuleHandle* module);
277 GpuModuleHandle* module);
283 GpuModuleHandle* module);
288 static bool GetModuleFunction(GpuContext* context, GpuModuleHandle module,
296 static bool GetModuleSymbol(GpuContext* context, GpuModuleHandle module,
304 static void UnloadModule(GpuContext* context, GpuModuleHandle module);
|
| D | gpu_types.h | 55 using GpuModuleHandle = hipModule_t; variable
73 using GpuModuleHandle = CUmodule;
|