Home
last modified time | relevance | path

Searched refs:_PyTreeCodeGen (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/export/
H A D_unlift.py11 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
127 ) -> _PyTreeCodeGen:
147 return _PyTreeCodeGen(
H A Dexported_program.py29 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
334 mod.graph._codegen = _PyTreeCodeGen(
H A D_trace.py77 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.py12 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
39 gm.graph._codegen = _PyTreeCodeGen(
H A D_capture.py247 if isinstance(graph_module.graph._codegen, torch.fx.graph._PyTreeCodeGen):
/aosp_15_r20/external/pytorch/torch/fx/
H A D_symbolic_trace.py32 from .graph import _PyTreeCodeGen, _PyTreeInfo, Graph
670 self.graph._codegen = _PyTreeCodeGen(
678 assert isinstance(self.graph._codegen, _PyTreeCodeGen)
H A Dgraph_module.py20 from .graph import _custom_builtins, _is_from_torch, _PyTreeCodeGen, Graph, PythonCode
761 if isinstance(self._graph._codegen, _PyTreeCodeGen):
H A Dgraph.py721 class _PyTreeCodeGen(CodeGen): class
/aosp_15_r20/external/pytorch/torch/_inductor/
H A D__init__.py63 if isinstance(gm.graph._codegen, torch.fx.graph._PyTreeCodeGen):
H A Dcompile_fx.py59 from ..fx.graph import _PyTreeCodeGen
1324 if isinstance(model_.graph._codegen, _PyTreeCodeGen):
/aosp_15_r20/external/pytorch/torch/_export/
H A D__init__.py44 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Deval_frame.py67 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
1215 new_graph.graph._codegen = _PyTreeCodeGen(