pytorch/torch/export
2024-10-08 22:30:24 +00:00
..
experimental Deprecate _preserve_ops and consolidate with decomp_table (#135080) 2024-09-15 17:01:58 +00:00
passes [export] Make move_to_device_pass function public (#134263) 2024-08-23 23:18:30 +00:00
__init__.py Create export_for_inference API and expose core_aten as public facing API (#135912) 2024-09-15 17:05:07 +00:00
_remove_auto_functionalized_pass.py Redesign custom op functionlaization for better re-inplace (#134409) 2024-09-04 17:08:58 +00:00
_remove_effect_tokens_pass.py
_safeguard.py
_swap.py [export] Move swap to a different file (#137134) 2024-10-06 04:28:18 +00:00
_trace.py Add original forward names to schema so that prettify pass works (#136887) 2024-10-08 04:21:02 +00:00
_tree_utils.py
_unlift.py Add original forward names to schema so that prettify pass works (#136887) 2024-10-08 04:21:02 +00:00
custom_obj.py
dynamic_shapes.py [export] clean up dynamic markers from tensors (#137230) 2024-10-04 06:50:45 +00:00
exported_program.py Add original forward names to schema so that prettify pass works (#136887) 2024-10-08 04:21:02 +00:00
graph_signature.py [export] refactor ExportGraphSignature construction (#134059) 2024-08-23 23:29:28 +00:00
unflatten.py Lift restriction on training IR for unflatten (#137470) 2024-10-08 22:30:24 +00:00