Skip to content

Commit fb86945

Browse files
authored
[ARM] fix fp16 transpose (#9272)
1 parent 0b1cf60 commit fb86945

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

lite/kernels/arm/transpose_compute.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -414,6 +414,7 @@ void transpose_mat(const lite_api::float16_t* din,
414414
#endif
415415
ptr_din_row += 4;
416416
}
417+
data_out_ptr0 = ptr_out + size_wh + 4 * height * remain_ww;
417418
for (int w = 0; w < remain_ww_rem; w++) {
418419
INIT_PTR_4(lite_api::float16_t, data_out_ptr0, height)
419420
INIT_PTR_A4(lite_api::float16_t)

0 commit comments

Comments
 (0)