batch size 0 support in ChannelShuffle DNNLOWP op (#26858)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26858
Handle batch size = 0 in ChannelShuffle operator
Test Plan: CI
Reviewed By: jianyuh
Differential Revision: D17591041
fbshipit-source-id: 63373aa752406c1f38401c3e93d8e1954ce7281e