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

[MSLITE] space to depth bug

This commit is contained in:
ling 2021-03-02 21:55:05 +08:00
parent 692d158f5c
commit 929fd16cf6
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);
}