projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: d1f3d85) | patch
OpInfo for `nn.functional.layer_norm` (#63276)
authorKushashwa Ravi Shrimali <kushashwaravishrimali@gmail.com>
Wed, 1 Sep 2021 15:48:25 +0000 (08:48 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 1 Sep 2021 16:31:45 +0000 (09:31 -0700)
commitd5bfdd3dac33dfa84e2a511fa79c4ad4e0e6b822
treeb61f2aa483bb120fd0d8aa7340fe611395ddaf68tree | snapshot
parentd1f3d85fd80de2166114af1b5e16070a6d33a898commit | diff
OpInfo for `nn.functional.layer_norm` (#63276)

Summary:
Please see https://github.com/facebookresearch/functorch/issues/78 and https://github.com/pytorch/pytorch/issues/54261.

Note:

* This PR also adds a reference test inspired by existing tests in `test_nn.py`.

cc: mruberry zou3519

Pull Request resolved: https://github.com/pytorch/pytorch/pull/63276

Reviewed By: ejguan

Differential Revision: D30452483

Pulled By: zou3519

fbshipit-source-id: 2578d01ca34e031668a41bd284db60c31ae1fba8
test/test_nn.py diff | blob | history
torch/testing/_internal/common_methods_invocations.py diff | blob | history
Domain: Machine Learning / ML Framework;