change activation position in reorder_data_bfyx_to_blocked_format kernel (#16307)
This commit is contained in:
1 parent
790f74c01c
commit
461cc2aee8
1 file changed
+2
-2
+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); \
|
||||
} \
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user