Home
last modified time | relevance | path

Searched refs:cur_bottom_grad_pos (Results 1 – 6 of 6) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A Ddeformable_im2col.h350 index_t cur_bottom_grad_pos = (c * height + cur_h + dy) * width + cur_w + dx; in deformable_col2im_cpu() local
353 grad_im[cur_bottom_grad_pos] += weight * cur_top_grad; in deformable_col2im_cpu()
H A Ddeformable_im2col.cuh373 index_t cur_bottom_grad_pos = (c * height + cur_h + dy) * width + cur_w + dx; in deformable_col2im_gpu_kernel() local
376 atomicAdd(grad_im + cur_bottom_grad_pos, weight * cur_top_grad); in deformable_col2im_gpu_kernel()
H A Dmodulated_deformable_im2col.cuh339 int cur_bottom_grad_pos = ((b * channels + c) * height + cur_h + dy) * width + cur_w + dx; in modulated_deformable_col2im_gpu_kernel() local
341 atomicAdd(grad_im + cur_bottom_grad_pos, weight * cur_top_grad); in modulated_deformable_col2im_gpu_kernel()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A Ddeformable_im2col.h350 index_t cur_bottom_grad_pos = (c * height + cur_h + dy) * width + cur_w + dx; in deformable_col2im_cpu() local
353 grad_im[cur_bottom_grad_pos] += weight * cur_top_grad; in deformable_col2im_cpu()
H A Ddeformable_im2col.cuh373 index_t cur_bottom_grad_pos = (c * height + cur_h + dy) * width + cur_w + dx; in deformable_col2im_gpu_kernel() local
376 atomicAdd(grad_im + cur_bottom_grad_pos, weight * cur_top_grad); in deformable_col2im_gpu_kernel()
H A Dmodulated_deformable_im2col.cuh339 int cur_bottom_grad_pos = ((b * channels + c) * height + cur_h + dy) * width + cur_w + dx; in modulated_deformable_col2im_gpu_kernel() local
341 atomicAdd(grad_im + cur_bottom_grad_pos, weight * cur_top_grad); in modulated_deformable_col2im_gpu_kernel()