Tugsbayasgalan Manlaibaatar 90e886aa6c Sanity check for non-strict (#121687) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/121687 Approved by: https://github.com/avikchaudhuri ghstack dependencies: #121652, #121678		2024-03-12 18:21:32 +00:00
..
__init__.py	[export] Remove WrapperModule. (#121042)	2024-03-05 18:10:22 +00:00
_remove_effect_tokens_pass.py	[export] Add effect token to export (#121424)	2024-03-09 02:43:26 +00:00
_safeguard.py	[export] Update error message for set_grad (#121666)	2024-03-12 16:41:45 +00:00
_trace.py	Sanity check for non-strict (#121687)	2024-03-12 18:21:32 +00:00
_tree_utils.py	[export] do not use tree_flatten_spec (#118608)	2024-01-30 19:14:04 +00:00
_unlift.py	[export] Add effect token to export (#121424)	2024-03-09 02:43:26 +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] Add effect token to export (#121424)	2024-03-09 02:43:26 +00:00
unflatten.py	[torch.export] Support is_compiling() flag for non-strict mode (#119602)	2024-02-29 05:52:51 +00:00