pytorch/torch/_export
Angela Yi ddbf1aab64 [export] Add dynamic_shapes to _export.aot_compile (#110101)
Summary: Following the new dynamic_shapes API (introduced in https://github.com/pytorch/pytorch/pull/108448), we will also add a dynamic_shapes API to _export.aot_compile

Test Plan: CI

Differential Revision: D49653815

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110101
Approved by: https://github.com/gmagogsfm
2023-09-27 04:10:22 +00:00
..
db Improve unbacked symint error msg (#107806) 2023-08-25 01:07:09 +00:00
pass_infra [export] ExportPassBase + view_copy pass (#100000) 2023-04-26 21:01:25 +00:00
passes [exir] Add lift constant tensors passes after aten_to_edge (#109382) 2023-09-19 01:34:58 +00:00
serde Revert "Disallow skipping dynamo (#109476)" 2023-09-25 20:20:50 +00:00
__init__.py [export] Add dynamic_shapes to _export.aot_compile (#110101) 2023-09-27 04:10:22 +00:00
constraints.py [export] Refactor constrain_as_value and constrain_as_size (#106591) 2023-08-15 05:41:43 +00:00
error.py Add experimental export() API (#100034) 2023-04-28 06:12:59 +00:00
exported_program.py [export] Update how we input kwargs (#109160) 2023-09-19 00:04:32 +00:00
pass_base.py [export] Fix autogenerated stacktrace (#108217) 2023-08-30 17:44:06 +00:00
utils.py Serialize pytree to json string (#106116) 2023-08-27 14:34:49 +00:00
verifier.py [export] Verifier for exported program (#109519) 2023-09-26 18:47:43 +00:00
wrappers.py python functionalization: support higher order ops (#108656) 2023-09-20 04:37:31 +00:00