torch/fx/__init__.pyi

from torch.fx._symbolic_trace import (
    symbolic_trace as symbolic_trace,
    Tracer as Tracer,
    wrap as wrap,
)
from torch.fx.graph import Graph as Graph
from torch.fx.graph_module import GraphModule as GraphModule
from torch.fx.interpreter import Interpreter as Interpreter, Transformer as Transformer
from torch.fx.node import (
    has_side_effect as has_side_effect,
    map_arg as map_arg,
    Node as Node,
)
from torch.fx.proxy import Proxy as Proxy
from torch.fx.subgraph_rewriter import replace_pattern as replace_pattern