Searched refs:module_node (Results 1 – 4 of 4) sorted by relevance
36 auto module_node = call_node->input(0)->node(); in GetCallNodeVariableName() local38 if (!module_node->hasAttribute(attr::name)) { in GetCallNodeVariableName()41 std::string module_name = module_node->s(attr::name); in GetCallNodeVariableName()42 if (module_node->inputs().empty()) { in GetCallNodeVariableName()47 auto parent_module_value = module_node->input(0); in GetCallNodeVariableName()
196 module_node = gm.graph.call_module(203 orig_outputs[0].replace_all_uses_with(module_node, propagate_meta=True)207 proxy_out = torch.fx.Proxy(module_node)[i].node # type: ignore[index]210 … module_node.meta["val"] = tuple(orig_output.meta.get("val", None) for orig_output in orig_outputs)
127 module_node = ast.parse(full_code, filename=full_path)128 walk_ast(module_node, modname, imports)
335 module_node = gast.parse(src)336 nodes = module_node.body