[release/2.8] Fix condition to enable NHWC Batchnorm#2599

jeffdaily merged 1 commit intorelease/2.8ROCm/pytorch:release/2.8from

dnikolae/fix_nhwc_bn_condition_2.8ROCm/pytorch:dnikolae/fix_nhwc_bn_condition_2.8Copy head branch name to clipboard