提交 c6dcb312 编写于 作者: M Michal Tarnowski

Fixed NaN propagation from channel padding

上级 fef913af
...@@ -83,21 +83,22 @@ __kernel void eltwise(OUT_OF_RANGE_PARAMS ...@@ -83,21 +83,22 @@ __kernel void eltwise(OUT_OF_RANGE_PARAMS
#endif #endif
#endif #endif
#if INPUT_TYPE == 1 || INPUT_TYPE == 4 #if ((INPUT_TYPE == 1 || INPUT_TYPE == 4) && \
#if ELTWISE_TYPE == 0 || ELTWISE_TYPE == 1 || ELTWISE_TYPE == 4 || \ (ELTWISE_TYPE == 0 || ELTWISE_TYPE == 1 || ELTWISE_TYPE == 4 || \
ELTWISE_TYPE == 5 || ELTWISE_TYPE == 8 || ELTWISE_TYPE == 9 ELTWISE_TYPE == 5 || ELTWISE_TYPE == 8 || ELTWISE_TYPE == 9)) || \
const int remain_channel = channel - 4 * chan_idx; ((INPUT_TYPE != 1 || INPUT_TYPE != 4) && \
if (remain_channel < 4) { (ELTWISE_TYPE == 3 || ELTWISE_TYPE == 9))
switch (remain_channel) { const int remain_channel = channel - 4 * chan_idx;
case 1: if (remain_channel < 4) {
out.y = 0; switch (remain_channel) {
case 2: case 1:
out.z = 0; out.y = 0;
case 3: case 2:
out.w = 0; out.z = 0;
} case 3:
out.w = 0;
} }
#endif }
#endif #endif
WRITE_IMAGET(output, (int2)(pos, hb), out); WRITE_IMAGET(output, (int2)(pos, hb), out);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册