Home
last modified time | relevance | path

Searched refs:data_offset_w_ptr (Results 1 – 8 of 8) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A Ddeformable_im2col.cuh252 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_im2col_gpu_kernel() local
254 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_im2col_gpu_kernel()
357 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_gpu_kernel() local
359 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_gpu_kernel()
488 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_coord_gpu_kernel() local
490 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_coord_gpu_kernel()
H A Dmodulated_deformable_im2col.cuh218 …const int data_offset_w_ptr = ((2 * (i * kernel_w + j) + 1) * height_col + h_col) * width_col + w_… in modulated_deformable_im2col_gpu_kernel() local
221 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in modulated_deformable_im2col_gpu_kernel()
321 …const int data_offset_w_ptr = ((2 * (i * kernel_w + j) + 1) * height_col + h_out) * width_col + w_… in modulated_deformable_col2im_gpu_kernel() local
324 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in modulated_deformable_col2im_gpu_kernel()
443 …const int data_offset_w_ptr = (((2 * (i * kernel_w + j) + 1) * height_col + h_out) * width_col + w… in modulated_deformable_col2im_coord_gpu_kernel() local
446 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in modulated_deformable_col2im_coord_gpu_kernel()
H A Ddeformable_im2col.h334 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_cpu() local
336 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_cpu()
452 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_coord_cpu() local
454 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_coord_cpu()
H A Dmodulated_deformable_im2col.h151 const int data_offset_w_ptr = ((2 * (i * kernel_w + j) + 1) * height_col + h_col) in Map() local
155 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in Map()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A Ddeformable_im2col.cuh252 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_im2col_gpu_kernel() local
254 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_im2col_gpu_kernel()
357 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_gpu_kernel() local
359 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_gpu_kernel()
488 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_coord_gpu_kernel() local
490 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_coord_gpu_kernel()
H A Dmodulated_deformable_im2col.cuh218 …const int data_offset_w_ptr = ((2 * (i * kernel_w + j) + 1) * height_col + h_col) * width_col + w_… in modulated_deformable_im2col_gpu_kernel() local
221 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in modulated_deformable_im2col_gpu_kernel()
321 …const int data_offset_w_ptr = ((2 * (i * kernel_w + j) + 1) * height_col + h_out) * width_col + w_… in modulated_deformable_col2im_gpu_kernel() local
324 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in modulated_deformable_col2im_gpu_kernel()
443 …const int data_offset_w_ptr = (((2 * (i * kernel_w + j) + 1) * height_col + h_out) * width_col + w… in modulated_deformable_col2im_coord_gpu_kernel() local
446 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in modulated_deformable_col2im_coord_gpu_kernel()
H A Ddeformable_im2col.h334 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_cpu() local
336 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_cpu()
452 const index_t data_offset_w_ptr = data_offset_h_ptr + height_col * width_col; in deformable_col2im_coord_cpu() local
454 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in deformable_col2im_coord_cpu()
H A Dmodulated_deformable_im2col.h151 const int data_offset_w_ptr = ((2 * (i * kernel_w + j) + 1) * height_col + h_col) in Map() local
155 const DType offset_w = data_offset_ptr[data_offset_w_ptr]; in Map()