pytorch/torch/export
Tugsbayasgalan (Tugsuu) Manlaibaatar f316c35a34 [export] Support preserving submodule callling convention in non-strict export (#117796)
Summary: Title

Test Plan: CI

Reviewed By: zhxchen17

Differential Revision: D52889236

Pull Request resolved: https://github.com/pytorch/pytorch/pull/117796
Approved by: https://github.com/angelayi
2024-01-19 17:16:45 +00:00
..
__init__.py [export] introduce WrapperModule (#117571) 2024-01-18 03:40:34 +00:00
_safeguard.py [RELAND] Error grad mode op in export API (#117420) 2024-01-13 21:36:29 +00:00
_trace.py [export] Support preserving submodule callling convention in non-strict export (#117796) 2024-01-19 17:16:45 +00:00
_unlift.py [HigherOrderOp] change signature of map_impl (#117161) 2024-01-13 02:50:46 +00:00
custom_obj.py [export] Add lifted constant obj to input (#116985) 2024-01-18 22:10:53 +00:00
dynamic_shapes.py [export][refactor][4/n] Make equality_constraints optional (#116233) 2024-01-05 00:50:52 +00:00
exported_program.py [export] Add lifted constant obj to input (#116985) 2024-01-18 22:10:53 +00:00
graph_signature.py Initial torchbind support in PT2 (#117697) 2024-01-19 06:28:20 +00:00
unflatten.py Revert "Add node meta value into UnflattenedModule (#117686)" 2024-01-18 17:46:38 +00:00
wrapper.py [export] introduce WrapperModule (#117571) 2024-01-18 03:40:34 +00:00