Fix layernorm ad formula on weight and bias (#18233)
authorWanchao Liang <wanchaol@users.noreply.github.com>
Wed, 3 Apr 2019 23:50:46 +0000 (16:50 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 3 Apr 2019 23:58:33 +0000 (16:58 -0700)
commit843e6234f5f87f281a487fd4f8434e07101ee3ed
treeb1a6499bbcfb619f00f6d90605046c2d0a422359
parent0512e4e32348f01dc2184031a7c4d4644a455ac3
Fix layernorm ad formula on weight and bias (#18233)

Summary:
Fix the layernorm formula when weight and bias passed in.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18233

Differential Revision: D14760375

Pulled By: wanchaol

fbshipit-source-id: d6bd3b137bc04c391aa5c24d021d1f811ba2a877
aten/src/ATen/native/Normalization.cpp
test/test_jit.py
torch/csrc/jit/symbolic_script.cpp