diff --git a/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc b/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc index 7e8f50d0de08888debfdd6259ea7258ebe30c88e..252b57551e52691037c3b748e683005f61f69d04 100644 --- a/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc +++ b/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc @@ -102,7 +102,8 @@ class ElementwiseTensorOpConverter : public OpConverter { int right_one_num = dims_x.nbDims - axis - dims_y.nbDims; nvinfer1::IShuffleLayer* reshape_layer; nvinfer1::ITensor* reshape_y_tensor; - if (left_one_num > 0 || right_one_num > 0) { + if (dims_x.nbDims != dims_y.nbDims && + (left_one_num > 0 || right_one_num > 0)) { if (engine_->with_dynamic_shape()) { auto* y_shape_tensor = Shape(Y); auto* new_y_shape_tensor = y_shape_tensor; diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 802a13ffd779299decfc3d622026c1b17c262a43..f9d227550854b0d759a950f4696dc9199f930266 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -2567,7 +2567,7 @@ struct SimpleOpTypeSetTeller : public Teller { return false; } } - if (op_type == "lookup_table") { + if (op_type == "lookup_table" || op_type == "lookup_table_v2") { if (!with_dynamic_shape) { VLOG(3) << "the lookup_table does not support " "static shape yet";