Searched refs:_make_graph_module (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_lazy_graph_module.py | 13 _make_graph_module,
223 gm1 = _make_graph_module(
229 gm2 = _make_graph_module(gm, gm.graph)
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | graph_module.py | 197 from ._lazy_graph_module import _make_graph_module
198 …gm = _make_graph_module(com, graph, class_name=graphmodule_cls_name, graph_module_cls=graph_module…
864 from ._lazy_graph_module import _make_graph_module
865 res = _make_graph_module(self, self.graph)
|
| H A D | interpreter.py | 3 from ._lazy_graph_module import _make_graph_module
520 return _make_graph_module(self.module, self.new_graph)
|
| H A D | _lazy_graph_module.py | 57 def _make_graph_module(*args, graph_module_cls=None, **kwargs): function
|
| H A D | _symbolic_trace.py | 34 from ._lazy_graph_module import _make_graph_module
1285 return _make_graph_module(tracer.root, graph, name)
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | partitioners.py | 379 fwd_module = fx._lazy_graph_module._make_graph_module(joint_module, fwd_graph)
380 bwd_module = fx._lazy_graph_module._make_graph_module(joint_module, bwd_graph)
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | output_graph.py | 28 from torch.fx._lazy_graph_module import _make_graph_module # type: ignore[attr-defined]
1331 gm = _make_graph_module(root, self.graph)
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | proxy_tensor.py | 1163 return fx._lazy_graph_module._make_graph_module(tracer.root, graph, name)
|