Browse Source

!14025 [MS][LITE][CPU] transpose bug fix

From: @lzkcode
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
pull/14025/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
b260430ab1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/nnacl/fp32/pack_fp32.c

+ 1
- 1
mindspore/lite/nnacl/fp32/pack_fp32.c View File

@@ -308,7 +308,7 @@ void PackNHWCToNCHWFp32(const void *src, void *dst, int batches, int plane, int
return;
}
task_end = (task_id + 1) == thread_count ? plane : MSMIN(plane, task_start + offset_hw);
hw8 = task_start + (task_end - task_start) >= offset_hw ? offset_hw : 0;
hw8 = task_start + ((task_end - task_start) >= offset_hw ? offset_hw : 0);
} else {
hw8 *= C8NUM;
}


Loading…
Cancel
Save