Browse Source

!12599 bug fix for fracz_3d to ncdhw trans in host

From: @liubuyu
Reviewed-by: @zhoufeng54,@kisnwang
Signed-off-by: @kisnwang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
f6731dcb0d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/common/trans.cc

+ 1
- 1
mindspore/ccsrc/common/trans.cc View File

@@ -1330,7 +1330,7 @@ bool FracZ3DToNcdhw(const FormatArgs &args, void *result) {
auto h = args.host_shape[3];
auto w = args.host_shape[4];
auto n0 = args.device_shape[1];
auto ni = args.device_shape[1];
auto ni = args.device_shape[2];
auto c0 = args.device_shape[3];
auto hw = h * w;
auto dhw = d * hw;


Loading…
Cancel
Save