pytorch/torch/_export/passes
Tugsbayasgalan Manlaibaatar 0a6c40faba Fix constant returning (#137993)
When the constants are used twice in the exported graph (second one is returned as output), the lifting constant pass doesn't account for the second one being the output. THis PR fixes that.

Differential Revision: [D64406108](https://our.internmc.facebook.com/intern/diff/D64406108/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/137993
Approved by: https://github.com/avikchaudhuri
2024-10-16 16:42:09 +00:00
..
__init__.py
_node_metadata_hook.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
add_runtime_assertions_for_constraints_pass.py Do not generate -1* in SymPy expressions when canonicalising (#128411) 2024-06-13 16:49:59 +00:00
collect_tracepoints_pass.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
constant_folding.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00
functionalize_side_effectful_ops_pass.py [export] Refactor ExportPassBase. (#116778) 2024-01-04 21:32:14 +00:00
lift_constants_pass.py Fix constant returning (#137993) 2024-10-16 16:42:09 +00:00
remove_runtime_assertions.py Flip default value for mypy disallow_untyped_defs [1/11] (#127838) 2024-06-08 18:16:33 +00:00
replace_autocast_with_hop_pass.py [export] Merge functions in replace set_grad/autocast with HOO (#132724) 2024-08-09 17:25:07 +00:00
replace_quantized_ops_with_standard_ops_pass.py [ts_converter][reland] Add support for LinearOpContext and Conv2dOpContext in quantization pass (#133622) 2024-08-16 01:55:45 +00:00
replace_set_grad_with_hop_pass.py [export] Merge functions in replace set_grad/autocast with HOO (#132724) 2024-08-09 17:25:07 +00:00
replace_view_ops_with_view_copy_ops_pass.py Remove _BLACK_LISTED_OPS (#131361) 2024-07-24 00:15:27 +00:00
replace_with_hop_pass_util.py [BE] typing for decorators - fx/_compatibility (part 1) (#134202) 2024-08-22 17:07:33 +00:00