projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 9137854) | patch
Enable calling caffe2 LayerNorm from PyTorch and JIT (#15243)
authorSebastian Messmer <messmer@fb.com>
Fri, 11 Jan 2019 00:06:26 +0000 (16:06 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 11 Jan 2019 00:22:18 +0000 (16:22 -0800)
commit8136c39b5e22a307f5aa50b1c1ca6a4a11283eee
treee90fd0f6dce251c8ff02257182fd30c7f2f93ae9tree | snapshot
parent913785445eadcd8de1b55a5329e4424847565626commit | diff
Enable calling caffe2 LayerNorm from PyTorch and JIT (#15243)

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

Register it as a custom JIT op.

Reviewed By: dzhulgakov

Differential Revision: D13473791

fbshipit-source-id: 0f7e72e3efc85a75060a7597fadaf0a8bd289651
c10/core/opschema/layer_norm.h diff | blob | history
caffe2/operators/layer_norm_op.cc diff | blob | history
tools/build_variables.py diff | blob | history
torch/CMakeLists.txt diff | blob | history
torch/csrc/jit/c10_ops/layer_norm.cpp [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;