pytorch/torch/export
Zhengxu Chen c51bdf5acf [export] Exporter API prototype. (#153205)
Summary: see inline code comments for documentation

Test Plan:
CI

buck2 test --flagfile fbcode//mode/opt fbcode//caffe2/test:test_export -- -r TestPackage

Differential Revision: D74426900

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153205
Approved by: https://github.com/tugsbayasgalan
2025-05-11 14:20:09 +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 [doc fix] fix torch export docs for preserve_module_call_signature (#151140) 2025-04-18 02:55:35 +00:00
_draft_export.py Rename register_fake_profile to unsafe_generate_fake_kernels (#151797) 2025-04-21 23:08:15 +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