Tugsbayasgalan Manlaibaatar bb31e3f57e Add original forward names to schema so that prettify pass works (#136887) ... When we run_decomp, we retrace if it is training IR. As a result, we do need to reliably store the oroiginal forward names when we run decomp. Differential Revision: [D63064453](https://our.internmc.facebook.com/intern/diff/D63064453/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/136887 Approved by: https://github.com/angelayi		2024-10-08 04:21:02 +00:00
..
__init__.py
aoti_schema.py	[aoti] Add initial custom op support (#127034)	2024-07-24 20:29:55 +00:00
dynamic_shapes.py	[export] dynamic_shapes serialization, load/dump (#134718)	2024-09-05 05:39:44 +00:00
schema_check.py	Flip default value for mypy disallow_untyped_defs [2/11] (#127839)	2024-06-08 18:23:08 +00:00
schema.py	Add original forward names to schema so that prettify pass works (#136887)	2024-10-08 04:21:02 +00:00
schema.yaml	Add original forward names to schema so that prettify pass works (#136887)	2024-10-08 04:21:02 +00:00
serialize.py	Add original forward names to schema so that prettify pass works (#136887)	2024-10-08 04:21:02 +00:00
union.py	Flip default value for mypy disallow_untyped_defs [2/11] (#127839)	2024-06-08 18:23:08 +00:00