* make shape inference of BatchNorm layout neutral * refactor to use the axis variable to do BatchNorm shape inference * refactor to use the axis variable to do BatchNorm shape inference * add unittest to the axis param for batch norm shape inference