Searched refs:cur_top_grad (Results 1 – 6 of 6) sorted by relevance
340 const DType cur_top_grad = data_col[index]; in deformable_col2im_cpu() local353 grad_im[cur_bottom_grad_pos] += weight * cur_top_grad; in deformable_col2im_cpu()
363 const DType cur_top_grad = data_col[index]; in deformable_col2im_gpu_kernel() local376 atomicAdd(grad_im + cur_bottom_grad_pos, weight * cur_top_grad); in deformable_col2im_gpu_kernel()
329 const DType cur_top_grad = data_col[index] * mask; in modulated_deformable_col2im_gpu_kernel() local341 atomicAdd(grad_im + cur_bottom_grad_pos, weight * cur_top_grad); in modulated_deformable_col2im_gpu_kernel()