Searched refs:shared_module (Results 1 – 5 of 5) sorted by relevance
48 pykms = shared_module('pykms',
69 XlaDebugInfoManager::Get()->RegisterModule(module_name_, shared_module(), in CpuExecutable()87 XlaDebugInfoManager::Get()->UnregisterModule(module_name_, shared_module(), in ~CpuExecutable()
72 XlaDebugInfoManager::Get()->RegisterModule(module_name_, shared_module(), in GpuExecutable()77 XlaDebugInfoManager::Get()->UnregisterModule(module_name_, shared_module(), in ~GpuExecutable()
353 std::shared_ptr<HloModule> shared_module() const { return hlo_module_; } in shared_module() function
2110 modules.push_back(local_exec->executable()->shared_module()); in GetHloModules()