transpose bug fix

4 years ago · 3b6dabc05b
parent 18e98c6a0b
commit 3b6dabc05b
1 changed files with 1 additions and 1 deletions
--- a/mindspore/lite/nnacl/fp32/pack_fp32.c
+++ b/mindspore/lite/nnacl/fp32/pack_fp32.c
@ -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;
  }