pytorch/torch/_export
Zhengxu Chen e9c240670f [sigmoid] Add canonicalized IR as an option. (#116758)
Summary: as title, the "canonical" flag is added to sigmoid serializer, so that we can optionally "normalize" the IR to give stable names and orders to IR nodes, which could help with the cases to compare IR definitions.

Test Plan: buck run @//mode/opt //aps_models/ads/config_model_authoring/stability:cli export-generated-module-state-command

Differential Revision: D52431965

Pull Request resolved: https://github.com/pytorch/pytorch/pull/116758
Approved by: https://github.com/avikchaudhuri
2024-01-24 03:11:25 +00:00
..
db [export] change exportdb to require torch.nn.Module (#117570) 2024-01-18 03:40:10 +00:00
pass_infra
passes Don't use private accessor on SymNode to get _expr (#118007) 2024-01-23 14:29:19 +00:00
serde [sigmoid] Add canonicalized IR as an option. (#116758) 2024-01-24 03:11:25 +00:00
__init__.py [AOTI] Fix a bug in the torch._export.aot_load API (#118039) 2024-01-23 14:54:02 +00:00
error.py
exported_program.py
non_strict_utils.py Initial torchbind support in PT2 (#117697) 2024-01-19 06:28:20 +00:00
pass_base.py
utils.py
verifier.py [export] Add lifted constant obj to input (#116985) 2024-01-18 22:10:53 +00:00
wrappers.py [export] Support preserving submodule callling convention in non-strict export (#117796) 2024-01-19 17:16:45 +00:00