projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: f5d5853) | patch
add channels last for GroupNorm (#49821)
authormingfeima <mingfei.ma@intel.com>
Tue, 24 Aug 2021 05:53:35 +0000 (22:53 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 24 Aug 2021 05:54:59 +0000 (22:54 -0700)
commit5b7cdc5a3ddb9a1a3d46d05b2925b5b4713b0025
tree4c01f2800a90c4de7d9cf055956ee1da1d72abf5tree | snapshot
parentf5d585391d13287250e85cbd55a17c5e0b8ac2a8commit | diff
add channels last for GroupNorm (#49821)

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

Test Plan: Imported from OSS

Reviewed By: ejguan

Differential Revision: D26007053

Pulled By: VitalyFedyunin

fbshipit-source-id: 34a48d5d3b66a159febf3c3d96748fbaba1b9e31
aten/src/ATen/native/cpu/group_norm_kernel.cpp diff | blob | history
aten/src/ATen/native/group_norm.cpp diff | blob | history
test/test_nn.py diff | blob | history
Domain: Machine Learning / ML Framework;