Optimize LayerNormOp (#17604)
authorXiaomeng Yang <yangxm@fb.com>
Sat, 9 Mar 2019 01:35:17 +0000 (17:35 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 9 Mar 2019 01:38:14 +0000 (17:38 -0800)
commit0fd1dc45c037ae806352e08a7b83eeacfad101a0
tree3364891daca9ecb378f6b81bd3c3fe1982fbd4fa
parent65b00aa5972e23b2a70aa60dec5125671a3d7153
Optimize LayerNormOp (#17604)

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

Optimize LayerNormOp

i-am-not-moving-c2-to-c10

Reviewed By: houseroad

Differential Revision: D14274175

fbshipit-source-id: a7aa263a1b0eb109682d2be99306e7b2cdcc0faf
caffe2/operators/layer_norm_op.cc
caffe2/operators/layer_norm_op.cu
caffe2/operators/layer_norm_op.h
caffe2/utils/math/reduce.cc