frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

move functionalize out of experimental namespace (#85742)

Browse Source 
Did a very quick sanity check - it looks like functorch docs don't get the nice preview link that pytofch-bot gives for normal pytorch docs, so I built locally and scanned `html/generated/functorch.functionalize.html`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/85742
Approved by: https://github.com/zou3519
This commit is contained in:
Brian Hirsh
2022-09-27 12:03:54 -07:00
committed by PyTorch MergeBot
parent 796da4df4d
commit 913f5784d7
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
functorch/__init__.py
View File

@ -15,7 +15,7 @@ from . import _C
# functorch transforms # functorch transforms
from ._src.vmap import vmap from ._src.vmap import vmap
from ._src.eager_transforms import ( from ._src.eager_transforms import (
grad, grad_and_value, vjp, jacrev, jvp, jacfwd, hessian, grad, grad_and_value, vjp, jacrev, jvp, jacfwd, hessian, functionalize
) )
from ._src.python_key import make_fx from ._src.python_key import make_fx

2
functorch/docs/source/experimental.rst
View File

@ -8,5 +8,3 @@ Experimental Function Transforms
.. autosummary:: .. autosummary::
:toctree: generated :toctree: generated
:nosignatures: :nosignatures:
functionalize

1
functorch/docs/source/functorch.rst
View File

@ -17,6 +17,7 @@ Function Transforms
jacrev jacrev
jacfwd jacfwd
hessian hessian
functionalize
Utilities for working with torch.nn.Modules Utilities for working with torch.nn.Modules
------------------------------------------- -------------------------------------------

3
functorch/experimental/__init__.py
View File

@ -1,4 +1,5 @@
from .batch_norm_replacement import replace_all_batch_norm_modules_ from .batch_norm_replacement import replace_all_batch_norm_modules_
# PyTorch forward-mode is not mature yet # PyTorch forward-mode is not mature yet
from .._src.eager_transforms import jvp, jacfwd, hessian, functionalize from .._src.eager_transforms import jvp, jacfwd, hessian
from .._src.vmap import chunk_vmap from .._src.vmap import chunk_vmap
from functorch import functionalize