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:24:28 +08:00
parent d8592efae2
commit 1639ce42af
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/spacetobatch_impl.cu
View File

@ -52,8 +52,8 @@ __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; 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 = idx_on * oc;
input_pos = (input_pos + idx_ic) * 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_ih + 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); input_pos = (input_pos + ((idx_iw + pad_lft) - ((idx_on / on) % block_num)) / block_num);
output[input_pos] = input[pos]; output[input_pos] = input[pos];
} }
return; return;