pytorch/torch/_export
Zhengxu Chen b62c0d96bc [export] Support user input mutation. [1/2] (#114496)
Summary:
Serialization not implemented yet. Will do in the next diff.

Resolving Github issues:
https://github.com/pytorch/pytorch/issues/112429
https://github.com/pytorch/pytorch/issues/114142

Test Plan:
buck2 run mode/opt caffe2/test:test_export -- -r test_export_
input_mutation

Differential Revision: D51556962

Pull Request resolved: https://github.com/pytorch/pytorch/pull/114496
Approved by: https://github.com/tugsbayasgalan
2023-11-27 04:53:38 +00:00
..
db [export] Support user input mutation. [1/2] (#114496) 2023-11-27 04:53:38 +00:00
pass_infra [export] ExportPassBase + view_copy pass (#100000) 2023-04-26 21:01:25 +00:00
passes [export][retry] Move lifted tensors out of state_dict (#113689) 2023-11-15 09:24:49 +00:00
serde Fixed an export problem when moving tensors to CPU during torch.export.save (#114029) 2023-11-23 21:17:43 +00:00
__init__.py [export] Support user input mutation. [1/2] (#114496) 2023-11-27 04:53:38 +00:00
error.py Add experimental export() API (#100034) 2023-04-28 06:12:59 +00:00
exported_program.py [export][retry] Move lifted tensors out of state_dict (#113689) 2023-11-15 09:24:49 +00:00
pass_base.py [HigherOrderOp] Move _map.py to _higher_order_ops (#111152) 2023-11-16 03:04:12 +00:00
unflatten.py [BE]: Remove useless lambdas (#113602) 2023-11-14 20:06:48 +00:00
utils.py Revert "[pytree] register pytree node type in both C++ pytree and Python pytree (#112111)" 2023-11-23 09:59:32 +00:00
verifier.py [export] Support user input mutation. [1/2] (#114496) 2023-11-27 04:53:38 +00:00
wrappers.py [HigherOrderOp] fallthrough some keys by default. (#110478) 2023-10-05 16:25:42 +00:00