Searched refs:_get_module_hierarchy (Results 1 – 2 of 2) sorted by relevance
38 def _get_module_hierarchy(node: torch.fx.Node) -> str: function73 fqn = _get_module_hierarchy(node)
497 def _get_module_hierarchy(mod: torch.nn.Module) -> Dict[str, str]: function1725 _EXPORT_MODULE_HIERARCHY = _get_module_hierarchy(mod)1860 _EXPORT_MODULE_HIERARCHY = _get_module_hierarchy(mod)