__init__.pyi 465 B

1234567891011
  1. from ._symbolic_trace import (
  2. symbolic_trace as symbolic_trace,
  3. Tracer as Tracer,
  4. wrap as wrap,
  5. )
  6. from .graph import Graph as Graph
  7. from .graph_module import GraphModule as GraphModule
  8. from .interpreter import Interpreter as Interpreter, Transformer as Transformer
  9. from .node import has_side_effect as has_side_effect, map_arg as map_arg, Node as Node
  10. from .proxy import Proxy as Proxy
  11. from .subgraph_rewriter import replace_pattern as replace_pattern