fix op_teller with_dynamic_shape judge bug (#45384)

9e0baf6e · Yuanle Liu · GitHub · 1f3d16ad · 9e0baf6e
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/fluid/inference/tensorrt/op_teller.cc paddle/fluid/inference/tensorrt/op_teller.cc +2 -2

未找到文件。
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -340,9 +340,9 @@ bool OpTeller::Tell(const framework::ir::Node* node,
    if (!with_dynamic_shape) {
      std::string X_name;
      auto inputs = desc.Inputs();
-      if (inputs.count("X")) {
+      if (inputs.count("X") && !desc.Input("X").empty()) {
        X_name = desc.Input("X")[0];
-      } else if (inputs.count("Input")) {
+      } else if (inputs.count("Input") && !desc.Input("Input").empty()) {
        X_name = desc.Input("Input")[0];
      }
      auto* block = desc.Block();