提交 ee82928e 编写于 作者: X xiebaiyuan 提交者: GitHub

[LITE][OPENCL][Image] fix concat 3dims issue ,success nanoyolo ,test=… (#3029)

* [LITE][OPENCL][Image] fix concat 3dims issue ,success nanoyolo ,test=develop

* [LITE][OPENCL][Image] fix concat 3dims issue ,success nanoyolo,remove debug codes ,test=develop
上级 5014c3ce
...@@ -53,6 +53,17 @@ class ConcatComputeImage : public KernelLite<TARGET(kOpenCL), ...@@ -53,6 +53,17 @@ class ConcatComputeImage : public KernelLite<TARGET(kOpenCL),
auto in_dims = inputs[0]->dims(); auto in_dims = inputs[0]->dims();
axis_size_ = out_dims[axis]; axis_size_ = out_dims[axis];
axis_ = axis; axis_ = axis;
if (out_dims.size() < 4) {
if (out_dims.size() - axis == 1) {
// width
width_ = out_dims[1]; // c
flag_ = 3;
} else {
// height
width_ = out_dims[0]; // n
flag_ = 2;
}
} else {
switch (axis_) { switch (axis_) {
case 0: case 0:
width_ = out_dims[2]; // h width_ = out_dims[2]; // h
...@@ -74,6 +85,8 @@ class ConcatComputeImage : public KernelLite<TARGET(kOpenCL), ...@@ -74,6 +85,8 @@ class ConcatComputeImage : public KernelLite<TARGET(kOpenCL),
default: default:
printf("this axis: %d does not support \n", axis_); printf("this axis: %d does not support \n", axis_);
} }
}
for (int i = 1; i < inputs.size(); i++) { for (int i = 1; i < inputs.size(); i++) {
auto dims = inputs[i]->dims(); auto dims = inputs[i]->dims();
// auto flag = CHECK_EQ_OR_FALSE(in_dims.size(), dims.size()); // auto flag = CHECK_EQ_OR_FALSE(in_dims.size(), dims.size());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册