pytorch/torch/_export
2025-05-13 23:30:13 +00:00
..
db PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
pass_infra PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
passes [export][function schema] support exporting hop with function schema argument (#152073) 2025-05-01 05:22:02 +00:00
serde GPU lowering uses aoti_call_delegate (#153282) 2025-05-13 23:23:27 +00:00
__init__.py PEP585: More UP006 fixes (#146392) 2025-02-20 06:18:13 +00:00
converter.py [export] refactor DimHints for type errors (#149424) 2025-03-19 18:51:07 +00:00
error.py
non_strict_utils.py [export] support functools.partial forward (non-strict) (#153408) 2025-05-13 23:30:13 +00:00
pass_base.py BE: Swap functorch --> torch._higher_order_ops (#152620) 2025-05-01 21:53:23 +00:00
tools.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
utils.py [export] Preserve custom metadata for tensor constants (#152241) 2025-04-30 00:30:35 +00:00
verifier.py Expand allowed_getattr_types_for_subgm to torch.Tensor (#150867) 2025-04-09 11:01:45 +00:00
wrappers.py Support subclass constructor capturing in export (#147014) 2025-03-16 18:19:19 +00:00