projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e4477fe) | patch
Export group norm as ATen and add test (#15569)
authorLu Fang <lufang@fb.com>
Thu, 27 Dec 2018 22:42:01 +0000 (14:42 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 27 Dec 2018 22:44:29 +0000 (14:44 -0800)
commitd63740bc3fc6dd79d2265a842934b224d5194cf6
treef80d711df73700a3f8ccecf93b294fc5a3a8ddd9tree | snapshot
parente4477feb157602883d7c940c2fe883375d39df2bcommit | diff
Export group norm as ATen and add test (#15569)

Summary:
Short term solution, export group norm as an ATen op to unblock users.
Long term will add GroupNorm to onnx.

Add an end to end test for this one.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15569

Differential Revision: D13554293

Pulled By: houseroad

fbshipit-source-id: b4974c9ea2a1b81338ca1e5c6747efe2715d7932
test/onnx/test_pytorch_onnx_caffe2.py diff | blob | history
torch/onnx/symbolic.py diff | blob | history
Domain: Machine Learning / ML Framework;