feat: Add TRT support for 3D(batch_norm_op and elementwise_add_op) (#36446)

paddle/fluid/inference/tensorrt/convert/batch_norm_op.cc

@@ -147,9 +147,10 @@ class BatchNormOpConverter : public OpConverter {
       X = expand_layer->getOutput(0);
     }
 
-    layer = TRT_ENGINE_ADD_LAYER(
-        engine_, Scale, *X, nvinfer1::ScaleMode::kCHANNEL, shift_weights.get(),
-        scale_weights.get(), power_weights.get());
+    layer = TRT_ENGINE_ADD_LAYER(engine_, ScaleNd, *X,
+                                 nvinfer1::ScaleMode::kCHANNEL,
+                                 shift_weights.get(), scale_weights.get(),
+                                 power_weights.get(), dynamic_shape_offset);
 
     auto output_name = op_desc.Output("Y").front();
     engine_->SetWeights(op_desc.Input("Bias").front(),

paddle/fluid/inference/tensorrt/convert/elementwise_op.cc

@@ -83,8 +83,8 @@ class ElementwiseWeightOpConverter : public OpConverter {
       }
       if (op_type_ == "add") {
         nvinfer1::IScaleLayer* scale_layer = TRT_ENGINE_ADD_LAYER(
-            engine_, Scale, *X, scale_mode, shift_weights.get(),
-            scale_weights.get(), power_weights.get());
+            engine_, ScaleNd, *X, scale_mode, shift_weights.get(),
+            scale_weights.get(), power_weights.get(), dynamic_shape_offset);
         layer = scale_layer;
       } else if (op_type_ == "mul") {
         nvinfer1::IScaleLayer* scale_layer = TRT_ENGINE_ADD_LAYER(