pytorch/torch/export
2023-12-20 01:43:39 +00:00
..
__init__.py [export][fix] Add back export strict argument (#115668) 2023-12-12 22:59:10 +00:00
_trace.py Support nn_module_stack in torch.export(strict=False) (#115454) 2023-12-20 01:43:39 +00:00
_unlift.py [export][reland][refactor][3/n] Move unlift to separate file (#115558) 2023-12-12 05:37:07 +00:00
dynamic_shapes.py [export][reland][refactor][1/n] Split dynamic shapes (#115556) 2023-12-12 05:36:41 +00:00
exported_program.py [export][reland] Remove runtime assertion pass (#115597) 2023-12-15 03:22:03 +00:00
graph_signature.py Reland "[export] Support user input mutation. [1/2]" (#114496) (#114596) 2023-11-27 20:19:04 +00:00