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

Fixed NaN propagation from channel padding

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