Xuehai Pan
|
199e07f108
|
[pytree][BE] update treespec num_children access (#116370)
Change `len(treespec.children_spes) -> treespec.num_children`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/116370
Approved by: https://github.com/Skylion007
|
2023-12-24 20:54:32 +00:00 |
|
suo
|
d2d129de65
|
[sigmoid] replace unflatten with upstream version (#115468)
as title
Differential Revision: [D52000213](https://our.internmc.facebook.com/intern/diff/D52000213/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115468
Approved by: https://github.com/zhxchen17
|
2023-12-22 00:56:19 +00:00 |
|
suo
|
b5c866db13
|
[export] Add FlatArgsAdapter to unflatten (#115467)
This is the final divergence between our internal/external unflatteners.
Differential Revision: [D52001135](https://our.internmc.facebook.com/intern/diff/D52001135/)
@diff-train-skip-merge
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115467
Approved by: https://github.com/zhxchen17
ghstack dependencies: #115466, #115795
|
2023-12-21 20:52:36 +00:00 |
|
suo
|
01ec3d1113
|
[export] upstream some final fixes to OSS unflatten (#115795)
as title
Differential Revision: [D52141387](https://our.internmc.facebook.com/intern/diff/D52141387/)
@diff-train-skip-merge
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115795
Approved by: https://github.com/zhxchen17
ghstack dependencies: #115466
|
2023-12-21 20:52:36 +00:00 |
|
suo
|
bc3ef1684e
|
[export] refactor unflatten.py to be a top-level API (#115466)
This is in preparation for the merging of the internal and external versions of
the unflattener. Unflatten needs to be its own API because we are adding more
options to it in forthcoming diffs.
Differential Revision: [D52001133](https://our.internmc.facebook.com/intern/diff/D52001133/)
@diff-train-skip-merge
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115466
Approved by: https://github.com/zhxchen17
|
2023-12-21 20:52:29 +00:00 |
|