Home
last modified time | relevance | path

Searched refs:PyCodegen (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Doutput_graph.py43 from .codegen import PyCodegen
489 cg = PyCodegen(self.root_tx)
1084 codegen = PyCodegen(tx, root)
1122 [PyCodegen(tx).create_store(var) for var in reversed(restore_vars)]
1126 pass1 = PyCodegen(tx, root, graph_output_var)
1130 pass2 = PyCodegen(
1159 [PyCodegen(tx).create_store(var) for var in reversed(restore_vars)]
1164 [PyCodegen(tx).create_delete(graph_output_var)]
1400 cg = PyCodegen(tx)
H A Dside_effects.py17 from .codegen import PyCodegen
407 def codegen_save_tempvars(self, cg: PyCodegen): argument
529 def codegen_update_mutated(self, cg: PyCodegen): argument
H A Dcodegen.py43 class PyCodegen: class
H A Dsymbolic_convert.py61 from .codegen import PyCodegen
650 cg = PyCodegen(self)
2851 cg = PyCodegen(self)
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbase.py281 from ..codegen import PyCodegen
283 cg = PyCodegen(tx)
H A Dtorch.py21 from ..codegen import PyCodegen
1095 cg = PyCodegen(tx)