Allow implementing either backward or vjp for Function (#63434)
authorAlban Desmaison <albandes@fb.com>
Mon, 23 Aug 2021 14:05:51 +0000 (07:05 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 23 Aug 2021 14:07:11 +0000 (07:07 -0700)
commitbafd875f743d93ccb3463676ea29101cae1760d7
treea2e11da9249adf0a024085e0f925198425b6dc47
parent726fd26b3e5ecf205569e59cf7cd5a6cbf4387a0
Allow implementing either backward or vjp for Function (#63434)

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

Test Plan: Imported from OSS

Reviewed By: ejguan

Differential Revision: D30431968

Pulled By: albanD

fbshipit-source-id: 0bb88664283486a9fd3364e6c3d79442a44625c2
test/test_autograd.py
torch/autograd/function.py