angelayi af62a70fab [export] Fix nn_module_stack in retracing (#121423) ... Fixes https://fb.workplace.com/groups/1075192433118967/permalink/1391916691446538/ Pull Request resolved: https://github.com/pytorch/pytorch/pull/121423 Approved by: https://github.com/zhxchen17		2024-03-08 00:34:11 +00:00
..
__init__.py	[export] Remove WrapperModule. (#121042)	2024-03-05 18:10:22 +00:00
_safeguard.py
_trace.py	[export] Fix nn_module_stack in retracing (#121423)	2024-03-08 00:34:11 +00:00
_tree_utils.py	[export] do not use tree_flatten_spec (#118608)	2024-01-30 19:14:04 +00:00
_unlift.py	[pt2 export] small fix on non_persistent buffer unlift (#120715)	2024-03-01 20:20:00 +00:00
custom_obj.py	[Export] Introduce class_fqn into CustomObjArgument (#118158)	2024-01-25 18:44:25 +00:00
dynamic_shapes.py	derived dim (#118729)	2024-02-28 19:48:32 +00:00
exported_program.py	[export][reland] Disable exported_program.__call__ (#120019)	2024-03-05 05:29:46 +00:00
graph_signature.py	[export] support non-persistent buffers (#118969)	2024-02-02 19:16:08 +00:00
unflatten.py	[torch.export] Support is_compiling() flag for non-strict mode (#119602)	2024-02-29 05:52:51 +00:00