Browse Source

!12806 [MSLITE] space to depth bug

From: @ling_qiao_min
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
bcf31414bb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/nnacl/base/space_to_depth_base.c

+ 1
- 1
mindspore/lite/nnacl/base/space_to_depth_base.c View File

@@ -47,7 +47,7 @@ int SpaceToDepthForNHWC(const void *input, void *output, const int *in_shape, co
size_t in_offset_w = in_offset_h + k * block_size * in_strides[2]; size_t in_offset_w = in_offset_h + k * block_size * in_strides[2];
size_t out_offset_w = out_offset_h + k * out_strides[2]; size_t out_offset_w = out_offset_h + k * out_strides[2];
for (int l = 0; l < block_size; ++l) { for (int l = 0; l < block_size; ++l) {
memcpy(int8_outoput_ptr + out_offset_w + l * block_size * in_strides[2] * data_size,
memcpy(int8_outoput_ptr + (out_offset_w + l * block_size * in_strides[2]) * data_size,
int8_input_ptr + (in_offset_w + l * in_strides[1]) * data_size, int8_input_ptr + (in_offset_w + l * in_strides[1]) * data_size,
block_size * in_strides[2] * data_size); block_size * in_strides[2] * data_size);
} }


Loading…
Cancel
Save