p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!12806 [MSLITE] space to depth bug 

From: @ling_qiao_min
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-03-03 09:18:41 +08:00 committed by Gitee
parent 968be1587d 929fd16cf6
commit bcf31414bb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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 out_offset_w = out_offset_h + k * out_strides[2];
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,
block_size * in_strides[2] * data_size);
}