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

spacetobatch

This commit is contained in:
zong_shuai 2021-08-24 17:22:59 +08:00
parent 5dfe1f0c1b
commit d8592efae2
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/ccsrc/backend/kernel_compiler/gpu/cuda_impl/spacetobatch_impl.cu
View File

@ -51,7 +51,7 @@ __global__ void SpaceToBatch(const size_t size, const T *input, const size_t in,
idx_on = (((idx_ih + pad_up) % block_num) * block_num + ((idx_iw + pad_lft) % block_num)) * in + idx_in;
input_pos = idx_on * oc;
input_pos = (input_pos + idx_oc) * oh;
input_pos = (input_pos + idx_ic) * oh;
input_pos = (input_pos + ((idx_oh + pad_up) - (idx_on / (on * block_num))) / block_num) * ow;
input_pos = (input_pos + ((idx_ow + pad_lft) - ((idx_on / on) % block_num)) / block_num);
output[input_pos] = input[pos];