pytorch/torch/export
2025-05-14 02:18:36 +00:00
..
experimental [export] Exporter API prototype. (#153205) 2025-05-11 14:20:09 +00:00
passes PEP585: More UP006 fixes (#146392) 2025-02-20 06:18:13 +00:00
__init__.py [export] Make draft_export public (#153219) 2025-05-14 02:18:36 +00:00
_draft_export.py [export] Make draft_export public (#153219) 2025-05-14 02:18:36 +00:00
_remove_auto_functionalized_pass.py remove allow-untyped-defs from torch/export/_remove_auto_functionalized_pass.py (#144230) 2025-01-06 22:23:19 +00:00
_remove_effect_tokens_pass.py Fakify torchbind objects in compile_fx and add tests for SigridTransformsInstanceTorchBind (#149529) 2025-03-21 18:58:28 +00:00
_safeguard.py
_swap.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
_trace.py [export] Ignore None buffers (#152571) 2025-05-01 18:18:16 +00:00
_tree_utils.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
_unlift.py Suppress more warnings (#149833) 2025-04-01 05:33:04 +00:00
_wrapper_utils.py [reland] Make export._trace._WrapperModule work in strict mode (#146919) (#151264) 2025-04-15 18:35:34 +00:00
custom_obj.py
custom_ops.py Fix subclass access custom op bug (#149698) 2025-03-21 19:42:56 +00:00
decomp_utils.py Re-land exclude upsample_bilinear2d.vec and nearest2d.vec from default export decomposition table (#147153) 2025-02-19 23:03:29 +00:00
dynamic_shapes.py Fix additional inputs to error on inconsistent constants (#151970) 2025-04-30 01:38:17 +00:00
exported_program.py [export][function schema] support exporting hop with function schema argument (#152073) 2025-05-01 05:22:02 +00:00
graph_signature.py pretty print graph signature (#149710) 2025-03-21 21:31:58 +00:00
unflatten.py [export] Unflatten None (#153000) 2025-05-08 16:40:13 +00:00