Home
last modified time | relevance | path

Searched refs:cur_top_grad (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.h340 const DType cur_top_grad = data_col[index]; in deformable_col2im_cpu() local
353 grad_im[cur_bottom_grad_pos] += weight * cur_top_grad; in deformable_col2im_cpu()
H A Ddeformable_im2col.cuh363 const DType cur_top_grad = data_col[index]; 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.cuh329 const DType cur_top_grad = data_col[index] * mask; 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.h340 const DType cur_top_grad = data_col[index]; in deformable_col2im_cpu() local
353 grad_im[cur_bottom_grad_pos] += weight * cur_top_grad; in deformable_col2im_cpu()
H A Ddeformable_im2col.cuh363 const DType cur_top_grad = data_col[index]; 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.cuh329 const DType cur_top_grad = data_col[index] * mask; 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()