optimize group_norm (#16216)
authorXiaomeng Yang <yangxm@fb.com>
Thu, 24 Jan 2019 07:55:06 +0000 (23:55 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 24 Jan 2019 07:57:45 +0000 (23:57 -0800)
commit60241e94b3e71f7637230648dcbbd068b71030b4
treeaaa86909a3fb49d3bf7507a0aed7652e98076668
parent8ab4d348f464fe9bf28bf6cbf518f97a65efae1b
optimize group_norm (#16216)

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

Optimize GroupNormOp

Reviewed By: houseroad

Differential Revision: D13754145

fbshipit-source-id: 650f64c81486c6c9d276f2e3325392d5838751ba
caffe2/operators/group_norm_op.cu