test=develop

Author: ceci3

paddle/fluid/operators/batch_norm_op.cc

@@ -68,16 +68,17 @@ void BatchNormOp::InferShape(framework::InferShapeContext *ctx) const {
   auto scale_dim = ctx->GetInputDim("Scale");
   auto bias_dim = ctx->GetInputDim("Bias");
 
+  PADDLE_ENFORCE_EQ(scale_dim.size(), 1UL);
+  PADDLE_ENFORCE_EQ(scale_dim.size(), 1UL);
+
   bool check = true;
   if ((!ctx->IsRuntime()) && (framework::product(scale_dim) <= 0 ||
                               framework::product(bias_dim) <= 0)) {
     check = false;
   }
 
   if (check) {
-    PADDLE_ENFORCE_EQ(scale_dim.size(), 1UL);
     PADDLE_ENFORCE_EQ(scale_dim[0], C);
-    PADDLE_ENFORCE_EQ(scale_dim.size(), 1UL);
     PADDLE_ENFORCE_EQ(scale_dim[0], C);
   }
   ctx->SetOutputDim("Y", x_dims);