Support Conv+BatchNorm fusion for 1d/3d (#29113)
Summary:
Support Conv+BatchNorm fusion for 1d/3d by being adaptive to number of dimensions (partially fixes https://github.com/pytorch/pytorch/issues/28757)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29113
Differential Revision: D18298248
Pulled By: soumith
fbshipit-source-id: 2fc75353aecc0e315c90e63476481acef6ebf784