ai-content-maker/.venv/Lib/site-packages/torch/fx/__init__.pyi

12 lines
465 B
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
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