Home
last modified time | relevance | path

Searched refs:module_handle (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h115 ModuleHandle *module_handle);
118 bool UnloadModule(ModuleHandle module_handle);
188 ModuleHandle module_handle = {});
192 const string &symbol_name, ModuleHandle module_handle = {});
541 bool GetSymbol(const string &symbol_name, ModuleHandle module_handle,
729 ModuleHandle module_handle) in ScopedModuleHandle() argument
730 : executor_(executor), module_handle_(module_handle) {} in ScopedModuleHandle()
772 const string &symbol_name, ModuleHandle module_handle) { in GetSymbol() argument
774 GetUntypedSymbol(symbol_name, module_handle); in GetSymbol()
Dstream_executor_pimpl.cc244 ModuleHandle *module_handle) { in LoadModule() argument
245 return implementation_->LoadModule(spec, module_handle); in LoadModule()
248 bool StreamExecutor::UnloadModule(ModuleHandle module_handle) { in UnloadModule() argument
249 return implementation_->UnloadModule(module_handle); in UnloadModule()
519 const string &symbol_name, ModuleHandle module_handle) { in GetUntypedSymbol() argument
524 if (GetSymbol(symbol_name, module_handle, &opaque, &bytes)) { in GetUntypedSymbol()
528 if (static_cast<bool>(module_handle)) { in GetUntypedSymbol()
533 reinterpret_cast<uintptr_t>(module_handle.id()), ")")); in GetUntypedSymbol()
543 ModuleHandle module_handle, void **mem, in GetSymbol() argument
545 return implementation_->GetSymbol(symbol_name, module_handle, mem, bytes); in GetSymbol()
Dstream_executor_internal.h187 ModuleHandle *module_handle) { in LoadModule() argument
190 virtual bool UnloadModule(ModuleHandle module_handle) { return false; } in UnloadModule() argument
284 virtual bool GetSymbol(const string &symbol_name, ModuleHandle module_handle, in GetSymbol() argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc202 se::ModuleHandle module_handle; in ResolveConstantGlobals() local
203 executor->LoadModule(module_spec, &module_handle); in ResolveConstantGlobals()
213 module_handle)); in ResolveConstantGlobals()
232 se::ScopedModuleHandle(executor, module_handle)); in ResolveConstantGlobals()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h69 ModuleHandle* module_handle) override;
70 bool UnloadModule(ModuleHandle module_handle) override;
198 bool GetSymbol(const string& symbol_name, ModuleHandle module_handle,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc116 bool GpuExecutor::UnloadModule(ModuleHandle module_handle) { in UnloadModule() argument
117 const char* gpu_binary = reinterpret_cast<const char*>(module_handle.id()); in UnloadModule()
364 ModuleHandle* module_handle) { in LoadModule() argument
376 *module_handle = ModuleHandle(const_cast<void*>( in LoadModule()
770 ModuleHandle module_handle, void** mem, in GetSymbol() argument
796 if (static_cast<bool>(module_handle)) { in GetSymbol()
797 auto it = gpu_binary_to_module_.find(module_handle.id()); in GetSymbol()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc361 ModuleHandle* module_handle) { in LoadModule() argument
372 *module_handle = ModuleHandle(const_cast<void *>( in LoadModule()
388 *module_handle = ModuleHandle(const_cast<void *>( in LoadModule()
396 bool GpuExecutor::UnloadModule(ModuleHandle module_handle) { in UnloadModule() argument
397 const char *gpu_binary = reinterpret_cast<const char *>(module_handle.id()); in UnloadModule()
889 ModuleHandle module_handle, void** mem, in GetSymbol() argument
900 if (static_cast<bool>(module_handle)) { in GetSymbol()
901 auto it = gpu_binary_to_module_.find(module_handle.id()); in GetSymbol()
/external/e2fsprogs/intl/
Drelocatable.c288 DllMain (HINSTANCE module_handle, DWORD event, LPVOID reserved) in DllMain() argument
297 if (!GetModuleFileName (module_handle, location, sizeof (location))) in DllMain()
/external/python/cpython3/Modules/clinic/
D_winapi.c.h519 _winapi_GetModuleFileName_impl(PyObject *module, HMODULE module_handle);
525 HMODULE module_handle; in _winapi_GetModuleFileName() local
527 if (!PyArg_Parse(arg, "" F_HANDLE ":GetModuleFileName", &module_handle)) { in _winapi_GetModuleFileName()
530 return_value = _winapi_GetModuleFileName_impl(module, module_handle); in _winapi_GetModuleFileName()
/external/python/cpython3/Modules/
D_winapi.c1233 _winapi_GetModuleFileName_impl(PyObject *module, HMODULE module_handle) in _winapi_GetModuleFileName_impl() argument
1239 result = GetModuleFileNameW(module_handle, filename, MAX_PATH); in _winapi_GetModuleFileName_impl()