Searched refs:_PyTreeCodeGen (Results 1 – 12 of 12) sorted by relevance
/aosp_15_r20/external/pytorch/torch/export/ |
H A D | _unlift.py | 11 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo 127 ) -> _PyTreeCodeGen: 147 return _PyTreeCodeGen(
|
H A D | exported_program.py | 29 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo 334 mod.graph._codegen = _PyTreeCodeGen(
|
H A D | _trace.py | 77 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo 1301 gm_torch_level.graph._codegen = _PyTreeCodeGen(
|
/aosp_15_r20/external/executorch/exir/capture/ |
H A D | _unlift.py | 12 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo 39 gm.graph._codegen = _PyTreeCodeGen(
|
H A D | _capture.py | 247 if isinstance(graph_module.graph._codegen, torch.fx.graph._PyTreeCodeGen):
|
/aosp_15_r20/external/pytorch/torch/fx/ |
H A D | _symbolic_trace.py | 32 from .graph import _PyTreeCodeGen, _PyTreeInfo, Graph 670 self.graph._codegen = _PyTreeCodeGen( 678 assert isinstance(self.graph._codegen, _PyTreeCodeGen)
|
H A D | graph_module.py | 20 from .graph import _custom_builtins, _is_from_torch, _PyTreeCodeGen, Graph, PythonCode 761 if isinstance(self._graph._codegen, _PyTreeCodeGen):
|
H A D | graph.py | 721 class _PyTreeCodeGen(CodeGen): class
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | __init__.py | 63 if isinstance(gm.graph._codegen, torch.fx.graph._PyTreeCodeGen):
|
H A D | compile_fx.py | 59 from ..fx.graph import _PyTreeCodeGen 1324 if isinstance(model_.graph._codegen, _PyTreeCodeGen):
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A D | __init__.py | 44 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | eval_frame.py | 67 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo 1215 new_graph.graph._codegen = _PyTreeCodeGen(
|