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

functionalization: add support for zero_()

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75913

Approved by: https://github.com/albanD
This commit is contained in:
Brian Hirsh
2022-04-25 12:01:17 -07:00
committed by PyTorch MergeBot
parent c8fd1bbc11
commit 7d44b3675b
11 changed files with 63 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tools/autograd/gen_python_functions.py
View File

@ -151,6 +151,7 @@ _SKIP_PYTHON_BINDINGS = [
"_has_same_storage_numel", # used for forward AD internals
"_reshape_alias",
"replace_", # only used by the functionalization pass, doesn't need to be exposed to python
"zero", # only used by the functionalization pass, doesn't need to be exposed to python
]
SKIP_PYTHON_BINDINGS = list(