pytorch/torch/_export
Shruthi GN ef9e89984c [pytorch] Support output types that are non tensors (#120804)
Summary:
per title
This is needed because some modules return None and non tensors as output

Test Plan: sandcastle?

Reviewed By: zhxchen17

Differential Revision: D54311609

Pull Request resolved: https://github.com/pytorch/pytorch/pull/120804
Approved by: https://github.com/zhxchen17
2024-02-29 02:49:10 +00:00
..
db Enable local_partial_types (#118467) 2024-01-28 13:38:22 +00:00
pass_infra [export] ExportPassBase + view_copy pass (#100000) 2023-04-26 21:01:25 +00:00
passes [export] preserve constant fqn (#120664) 2024-02-27 06:35:51 +00:00
serde [pytorch] Support output types that are non tensors (#120804) 2024-02-29 02:49:10 +00:00
__init__.py derived dim (#118729) 2024-02-28 19:48:32 +00:00
error.py Add experimental export() API (#100034) 2023-04-28 06:12:59 +00:00
exported_program.py [export] Remove CallSpec (#117671) 2024-02-08 17:19:03 +00:00
non_strict_utils.py Allow str inputs in non-strict tracing (#120536) 2024-02-28 23:56:30 +00:00
pass_base.py [HigherOrderOp] change signature of map_impl (#117161) 2024-01-13 02:50:46 +00:00
utils.py derived dim (#118729) 2024-02-28 19:48:32 +00:00
verifier.py [export] support non-persistent buffers (#118969) 2024-02-02 19:16:08 +00:00
wrappers.py [export] Use forward hooks to capture module signatures. (#120468) 2024-02-27 17:44:06 +00:00