projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 25e2578) | patch
Add forward AD support for custom Functions (#64061)
authorAlban Desmaison <albandes@fb.com>
Wed, 1 Sep 2021 20:34:48 +0000 (13:34 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 1 Sep 2021 21:33:09 +0000 (14:33 -0700)
commite322547fe6dd4f0ca9261a1ac2ae7095800b98a1
treeff18ba222d8bec14e18e84573c5a171c83003bd7tree | snapshot
parent25e2578967494ee88da66820d861c426eb7a742dcommit | diff
Add forward AD support for custom Functions (#64061)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64061

Test Plan: Imported from OSS

Reviewed By: soulitzer

Differential Revision: D30640868

Pulled By: albanD

fbshipit-source-id: b0e6610430a879074d6d5306443772fc154b431f
test/test_autograd.py diff | blob | history
torch/autograd/function.py diff | blob | history
torch/csrc/autograd/custom_function.cpp diff | blob | history
torch/csrc/autograd/custom_function.h diff | blob | history
torch/csrc/autograd/python_function.cpp diff | blob | history
Domain: Machine Learning / ML Framework;