未验证 提交 4ea34278 编写于 作者: Z zlsh80826 提交者: GitHub

[Paddle-TRT] support batch axis concatenation when using dynamic shape (#31627)

* support batch axis concatenation when using dynamic shape

* opteller can't return true early, or some test will not be executed
上级 d4282ea9
......@@ -159,7 +159,11 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
return false;
} else {
int axis = BOOST_GET_CONST(int, desc.GetAttr("axis"));
if (axis <= 0) return false;
if (with_dynamic_shape) {
if (axis < 0) return false;
} else {
if (axis <= 0) return false;
}
}
}
if (op_type == "transpose2" || op_type == "transpose") {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册