xref: /aosp_15_r20/external/pytorch/torch/fx/__init__.pyi (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1from torch.fx._symbolic_trace import (
2    symbolic_trace as symbolic_trace,
3    Tracer as Tracer,
4    wrap as wrap,
5)
6from torch.fx.graph import Graph as Graph
7from torch.fx.graph_module import GraphModule as GraphModule
8from torch.fx.interpreter import Interpreter as Interpreter, Transformer as Transformer
9from torch.fx.node import (
10    has_side_effect as has_side_effect,
11    map_arg as map_arg,
12    Node as Node,
13)
14from torch.fx.proxy import Proxy as Proxy
15from torch.fx.subgraph_rewriter import replace_pattern as replace_pattern
16