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

Author: feng_shuai

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(