change activation position in reorder_data_bfyx_to_blocked_format kernel (#16307)

This commit is contained in:
Wilson Seok authored and GitHub committed 2023-03-16 17:48:23 +09:00
1 parent 790f74c01c
commit 461cc2aee8
1 file changed
+2 -2
@@ -22,7 +22,7 @@
INPUTVTYPE read_data = AS_INPUTVTYPE(VLOAD(0, input + input_idx)); \
unroll_for (uint lw = 0; lw < inner; ++lw) { \
const uint dst = local_buf_offset + lw; \
transpose_buf[dst][lh] = ACTIVATION(read_data[lw], ACTIVATION_PARAMS); \
transpose_buf[dst][lh] = read_data[lw]; \
} \
}
@@ -34,7 +34,7 @@
#define FUNC_WRITE(inner, outer) unroll_for (uint lw = 0; lw < outer; ++lw) { \
const uint output_idx = output_idx_tile + (lw * x_pitch); \
unroll_for (uint i = 0; i < inner; ++i) { \
output[output_idx + i] = TO_OUTPUT_TYPE(transpose_buf[local_buf_offset + lw][i]); \
output[output_idx + i] = ACTIVATION(TO_OUTPUT_TYPE(transpose_buf[local_buf_offset + lw][i]), ACTIVATION_PARAMS); \
} \
}