Searched refs:num_spatial_axes (Results 1 – 10 of 10) sorted by relevance
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | im2col.h | 154 int num_spatial_axes = kernel_shape.ndim(); 157 for (index_t i = 0; i < num_spatial_axes; ++i) { 165 for (index_t i = 0; i < num_spatial_axes; ++i) { 169 std::vector<index_t> d_offset(num_spatial_axes, 0); 170 std::vector<index_t> d_iter(num_spatial_axes, 0); 174 for (int d_i = static_cast<int>(num_spatial_axes) - 1; d_i >= 0; --d_i) { 175 if (d_i < static_cast<int>(num_spatial_axes) - 1) { 186 for (index_t d_i = 0; d_i < num_spatial_axes; ++d_i) { 208 for (int d_i = static_cast<int>(num_spatial_axes) - 1; d_i >= 0; --d_i) { 321 int num_spatial_axes = kernel_shape.ndim(); in col2im() local [all …]
|
H A D | im2col.cuh | 289 index_t num_spatial_axes = kernel_shape.ndim(); in im2col() local 290 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in im2col() 293 switch (num_spatial_axes) { in im2col() 318 << num_spatial_axes << " spatial axes"; in im2col() 474 index_t num_spatial_axes = kernel_shape.ndim(); in col2im() local 477 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in col2im() 479 switch (num_spatial_axes) { in col2im() 511 << num_spatial_axes << " spatial axes"; in col2im()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | im2col.h | 154 int num_spatial_axes = kernel_shape.ndim(); 157 for (index_t i = 0; i < num_spatial_axes; ++i) { 165 for (index_t i = 0; i < num_spatial_axes; ++i) { 169 std::vector<index_t> d_offset(num_spatial_axes, 0); 170 std::vector<index_t> d_iter(num_spatial_axes, 0); 174 for (int d_i = static_cast<int>(num_spatial_axes) - 1; d_i >= 0; --d_i) { 175 if (d_i < static_cast<int>(num_spatial_axes) - 1) { 186 for (index_t d_i = 0; d_i < num_spatial_axes; ++d_i) { 208 for (int d_i = static_cast<int>(num_spatial_axes) - 1; d_i >= 0; --d_i) { 321 int num_spatial_axes = kernel_shape.ndim(); in col2im() local [all …]
|
H A D | im2col.cuh | 289 index_t num_spatial_axes = kernel_shape.ndim(); in im2col() local 290 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in im2col() 293 switch (num_spatial_axes) { in im2col() 318 << num_spatial_axes << " spatial axes"; in im2col() 474 index_t num_spatial_axes = kernel_shape.ndim(); in col2im() local 477 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in col2im() 479 switch (num_spatial_axes) { in col2im() 511 << num_spatial_axes << " spatial axes"; in col2im()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/ |
H A D | deformable_im2col.cuh | 300 const int num_spatial_axes = kernel_shape.ndim(); in deformable_im2col() local 301 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in deformable_im2col() 305 switch (num_spatial_axes) { in deformable_im2col() 320 << num_spatial_axes << " spatial axes"; in deformable_im2col() 410 const int num_spatial_axes = kernel_shape.ndim(); in deformable_col2im() local 415 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in deformable_col2im() 417 switch (num_spatial_axes) { in deformable_col2im() 435 << num_spatial_axes << " spatial axes"; in deformable_col2im() 536 const int num_spatial_axes = kernel_shape.ndim(); in deformable_col2im_coord() local 543 switch (num_spatial_axes) { in deformable_col2im_coord() [all …]
|
H A D | modulated_deformable_im2col.cuh | 266 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_im2col() local 267 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in modulated_deformable_im2col() 271 switch (num_spatial_axes) { in modulated_deformable_im2col() 283 << num_spatial_axes << " spatial axes"; in modulated_deformable_im2col() 370 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_col2im() local 375 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in modulated_deformable_col2im() 377 switch (num_spatial_axes) { in modulated_deformable_col2im() 392 << num_spatial_axes << " spatial axes"; in modulated_deformable_col2im() 491 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_col2im_coord() local 497 switch (num_spatial_axes) { in modulated_deformable_col2im_coord() [all …]
|
H A D | modulated_deformable_im2col.h | 199 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_im2col() local 203 if (2 == num_spatial_axes) { in modulated_deformable_im2col()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/ |
H A D | deformable_im2col.cuh | 300 const int num_spatial_axes = kernel_shape.ndim(); in deformable_im2col() local 301 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in deformable_im2col() 305 switch (num_spatial_axes) { in deformable_im2col() 320 << num_spatial_axes << " spatial axes"; in deformable_im2col() 410 const int num_spatial_axes = kernel_shape.ndim(); in deformable_col2im() local 415 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in deformable_col2im() 417 switch (num_spatial_axes) { in deformable_col2im() 435 << num_spatial_axes << " spatial axes"; in deformable_col2im() 536 const int num_spatial_axes = kernel_shape.ndim(); in deformable_col2im_coord() local 543 switch (num_spatial_axes) { in deformable_col2im_coord() [all …]
|
H A D | modulated_deformable_im2col.cuh | 266 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_im2col() local 267 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in modulated_deformable_im2col() 271 switch (num_spatial_axes) { in modulated_deformable_im2col() 283 << num_spatial_axes << " spatial axes"; in modulated_deformable_im2col() 370 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_col2im() local 375 CHECK_LT(num_spatial_axes, mshadow::cuda::kBaseThreadNum); in modulated_deformable_col2im() 377 switch (num_spatial_axes) { in modulated_deformable_col2im() 392 << num_spatial_axes << " spatial axes"; in modulated_deformable_col2im() 491 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_col2im_coord() local 497 switch (num_spatial_axes) { in modulated_deformable_col2im_coord() [all …]
|
H A D | modulated_deformable_im2col.h | 199 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_im2col() local 203 if (2 == num_spatial_axes) { in modulated_deformable_im2col()
|