/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/ |
H A D | modulated_deformable_im2col.cuh | 191 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 D | modulated_deformable_im2col.h | 118 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 D | deformable_im2col.h | 223 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 D | deformable_im2col.cuh | 297 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 D | modulated_deformable_im2col.cuh | 191 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 D | modulated_deformable_im2col.h | 118 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 D | deformable_im2col.h | 223 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 D | deformable_im2col.cuh | 297 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()
|