Viewing File: /home/ubuntu/combine_ai/combine/lib/python3.10/site-packages/torch/fx/__init__.pyi

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