未验证 提交 9e0baf6e 编写于 作者: Y Yuanle Liu 提交者: GitHub

fix op_teller with_dynamic_shape judge bug (#45384)

上级 1f3d16ad
...@@ -340,9 +340,9 @@ bool OpTeller::Tell(const framework::ir::Node* node, ...@@ -340,9 +340,9 @@ bool OpTeller::Tell(const framework::ir::Node* node,
if (!with_dynamic_shape) { if (!with_dynamic_shape) {
std::string X_name; std::string X_name;
auto inputs = desc.Inputs(); auto inputs = desc.Inputs();
if (inputs.count("X")) { if (inputs.count("X") && !desc.Input("X").empty()) {
X_name = desc.Input("X")[0]; 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]; X_name = desc.Input("Input")[0];
} }
auto* block = desc.Block(); auto* block = desc.Block();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册