pytorch/torch/export
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