Home
last modified time | relevance | path

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

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A Dmodulated_deformable_im2col.cuh191 const int batch_size, const int num_channels, const int deformable_group, in modulated_deformable_im2col_gpu_kernel() argument
264 const uint32_t deformable_group, DType* data_col) { in modulated_deformable_im2col() argument
268 index_t channel_per_deformable_group = im_shape[1] / deformable_group; in modulated_deformable_im2col()
301 const int batch_size, const int deformable_group, in modulated_deformable_col2im_gpu_kernel() argument
368 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im() argument
372 index_t channel_per_deformable_group = im_shape[1] / deformable_group; in modulated_deformable_col2im()
387 col_shape[1], deformable_group, col_shape[2], col_shape[3], grad_im, req); in modulated_deformable_col2im()
411 const int batch_size, const int offset_channels, const int deformable_group, in modulated_deformable_col2im_coord_gpu_kernel() argument
489 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im_coord() argument
493 index_t channel_per_deformable_group = col_shape[0] / deformable_group; in modulated_deformable_col2im_coord()
[all …]
H A Dmodulated_deformable_im2col.h118 const int batch_size, const int num_channels, const int deformable_group, in Map()
140 + (b_col * deformable_group + deformable_group_index) * 2 in Map()
144 + (b_col * deformable_group + deformable_group_index) * kernel_h in Map()
197 const uint32_t deformable_group, DType* data_col) { in modulated_deformable_im2col() argument
200 index_t channel_per_deformable_group = im_shape[1] / deformable_group; in modulated_deformable_im2col()
208 channel_per_deformable_group, col_shape[1], im_shape[1], deformable_group, in modulated_deformable_im2col()
235 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im() argument
262 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im_coord() argument
H A Ddeformable_im2col.h223 const index_t deformable_group, in deformable_im2col_cpu() argument
228 const index_t channel_per_group = channels / deformable_group; in deformable_im2col_cpu()
282 const index_t deformable_group, in deformable_im2col() argument
291 deformable_group, in deformable_im2col()
312 const index_t deformable_group, in deformable_col2im_cpu() argument
315 index_t channel_per_group = channels / deformable_group; in deformable_col2im_cpu()
385 const index_t deformable_group, in deformable_col2im() argument
393 deformable_group, in deformable_col2im()
416 const index_t deformable_group, in deformable_col2im_coord_cpu() argument
498 const index_t deformable_group, in deformable_col2im_coord() argument
[all …]
H A Ddeformable_im2col.cuh297 const index_t deformable_group, in deformable_im2col() argument
302 index_t channel_per_group = im_shape[1] / deformable_group; in deformable_im2col()
408 const index_t deformable_group, in deformable_col2im() argument
412 index_t channel_per_group = im_shape[1] / deformable_group; in deformable_col2im()
534 const index_t deformable_group, in deformable_col2im_coord() argument
538 kernel_shape[0] * kernel_shape[1] * deformable_group; in deformable_col2im_coord()
539 index_t channel_per_group = col_shape[0] / deformable_group; in deformable_col2im_coord()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A Dmodulated_deformable_im2col.cuh191 const int batch_size, const int num_channels, const int deformable_group, in modulated_deformable_im2col_gpu_kernel() argument
264 const uint32_t deformable_group, DType* data_col) { in modulated_deformable_im2col() argument
268 index_t channel_per_deformable_group = im_shape[1] / deformable_group; in modulated_deformable_im2col()
301 const int batch_size, const int deformable_group, in modulated_deformable_col2im_gpu_kernel() argument
368 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im() argument
372 index_t channel_per_deformable_group = im_shape[1] / deformable_group; in modulated_deformable_col2im()
387 col_shape[1], deformable_group, col_shape[2], col_shape[3], grad_im, req); in modulated_deformable_col2im()
411 const int batch_size, const int offset_channels, const int deformable_group, in modulated_deformable_col2im_coord_gpu_kernel() argument
489 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im_coord() argument
493 index_t channel_per_deformable_group = col_shape[0] / deformable_group; in modulated_deformable_col2im_coord()
[all …]
H A Dmodulated_deformable_im2col.h118 const int batch_size, const int num_channels, const int deformable_group, in Map()
140 + (b_col * deformable_group + deformable_group_index) * 2 in Map()
144 + (b_col * deformable_group + deformable_group_index) * kernel_h in Map()
197 const uint32_t deformable_group, DType* data_col) { in modulated_deformable_im2col() argument
200 index_t channel_per_deformable_group = im_shape[1] / deformable_group; in modulated_deformable_im2col()
208 channel_per_deformable_group, col_shape[1], im_shape[1], deformable_group, in modulated_deformable_im2col()
235 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im() argument
262 const TShape& dilation, const uint32_t deformable_group, in modulated_deformable_col2im_coord() argument
H A Ddeformable_im2col.h223 const index_t deformable_group, in deformable_im2col_cpu() argument
228 const index_t channel_per_group = channels / deformable_group; in deformable_im2col_cpu()
282 const index_t deformable_group, in deformable_im2col() argument
291 deformable_group, in deformable_im2col()
312 const index_t deformable_group, in deformable_col2im_cpu() argument
315 index_t channel_per_group = channels / deformable_group; in deformable_col2im_cpu()
385 const index_t deformable_group, in deformable_col2im() argument
393 deformable_group, in deformable_col2im()
416 const index_t deformable_group, in deformable_col2im_coord_cpu() argument
498 const index_t deformable_group, in deformable_col2im_coord() argument
[all …]
H A Ddeformable_im2col.cuh297 const index_t deformable_group, in deformable_im2col() argument
302 index_t channel_per_group = im_shape[1] / deformable_group; in deformable_im2col()
408 const index_t deformable_group, in deformable_col2im() argument
412 index_t channel_per_group = im_shape[1] / deformable_group; in deformable_col2im()
534 const index_t deformable_group, in deformable_col2im_coord() argument
538 kernel_shape[0] * kernel_shape[1] * deformable_group; in deformable_col2im_coord()
539 index_t channel_per_group = col_shape[0] / deformable_group; in deformable_col2im_coord()