pytorch/caffe2/python/helpers
Xiaomeng Yang 271f005eeb Add elementwise_affine for LayerNormGradientOp (#19982)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19982

Add elementwise_affine for LayerNormGradientOp

Reviewed By: houseroad

Differential Revision: D15157493

fbshipit-source-id: 7465f2c1d4df4649b4903b93483c4861e9c7afa9
2019-05-03 15:33:46 -07:00
..
__init__.py
algebra.py
arg_scope.py
array_helpers.py
control_ops.py
conv.py
db_input.py
dropout.py
elementwise_linear.py
fc.py
nonlinearity.py
normalization.py
pooling.py
tools.py
train.py