pytorch/torch/export
2024-11-22 00:05:05 +00:00
..
experimental Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
passes Fixing issue in move pass for copying Parameter (#138855) 2024-10-25 17:57:27 +00:00
__init__.py
_draft_export.py [draft export] generate fake outputs when real tensor prop finds mismatches (#139766) 2024-11-21 08:01:09 +00:00
_remove_auto_functionalized_pass.py
_remove_effect_tokens_pass.py
_safeguard.py
_swap.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
_trace.py Change export IR to non-functional pre-dispatch IR (#139511) 2024-11-20 21:47:55 +00:00
_tree_utils.py [export] Add setattr for ep.example_inputs (#140990) 2024-11-20 02:49:20 +00:00
_unlift.py [export] Add setattr for ep.example_inputs (#140990) 2024-11-20 02:49:20 +00:00
custom_obj.py
decomp_utils.py
dynamic_shapes.py make DimHints compatible with Dims (#138490) 2024-10-22 07:43:48 +00:00
exported_program.py [export] fix loss_output in joint graph signature (#140974) 2024-11-21 23:57:07 +00:00
graph_signature.py [export] fix loss_output in joint graph signature (#140974) 2024-11-21 23:57:07 +00:00
unflatten.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00