Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 75a2f9d5 编写于 作者: W wenbin 提交者： GitHub
浏览文件
操作

fix concat axis bug (#50951) 

* fix concat bug

* recommit for ci
上级 6c471ed0
显示空白变更内容
内联 并排
Showing with 5 addition and 4 deletion
paddle/fluid/inference/tensorrt/convert/concat_op.cc
浏览文件 @ 75a2f9d5
...@@ -45,10 +45,11 @@ class ConcatOpConverter : public OpConverter { ...@@ -45,10 +45,11 @@ class ConcatOpConverter : public OpConverter {
itensors.push_back(engine_->GetITensor(input_name)); itensors.push_back(engine_->GetITensor(input_name));
} }
int axis = PADDLE_GET_CONST(int, op_desc.GetAttr("axis")); int axis = PADDLE_GET_CONST(int, op_desc.GetAttr("axis"));
if (axis == -1) { if (axis < 0) {
axis = (engine_->GetITensor(op_desc.Input("X").front())->getDimensions()) axis = engine_->GetITensor(op_desc.Input("X").front())
.nbDims - ->getDimensions()
1; .nbDims +
axis;
} else { } else {
if (!engine_->with_dynamic_shape()) { if (!engine_->with_dynamic_shape()) {
axis = axis - 1; // Remove batch dim axis = axis - 1; // Remove batch dim
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或