Expose ChannelShuffle (#46000)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/45999
Also small fix for caffe2 counterpart
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46000
Reviewed By: mruberry
Differential Revision: D24185855
Pulled By: ngimel
fbshipit-source-id: c5d599bb8100b86b81c6901f1b8b8baefc12cb16