Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 331
Star 4
Fork 1
P
Paddle-Lite
提交 429309fb 编写于 作者: D dolphin8
浏览文件
操作

concat

上级 3b4ba914
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
metal/paddle-mobile/paddle-mobile/Operators/Kernels/metal/ConcatKernel.inc.metal
浏览文件 @ 429309fb
...@@ -65,7 +65,7 @@ kernel void FUNC(concat, R, N, VV, P)(texture2d_array<P, access::read> in0 [[tex ...@@ -65,7 +65,7 @@ kernel void FUNC(concat, R, N, VV, P)(texture2d_array<P, access::read> in0 [[tex
} }
k -= cp.vdim[j]; k -= cp.vdim[j];
} }
if (k > cp.vdim[N-1]) { if (j == N) {
continue; continue;
} }
int ta = cp.odim[cp.axis]; int ta = cp.odim[cp.axis];
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或