Home
last modified time | relevance | path

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 Dim2col.h154 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 Dim2col.cuh289 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 Dim2col.h154 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 Dim2col.cuh289 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 Ddeformable_im2col.cuh300 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 Dmodulated_deformable_im2col.cuh266 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 Dmodulated_deformable_im2col.h199 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 Ddeformable_im2col.cuh300 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 Dmodulated_deformable_im2col.cuh266 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 Dmodulated_deformable_im2col.h199 index_t num_spatial_axes = kernel_shape.ndim(); in modulated_deformable_im2col() local
203 if (2 == num_spatial_axes) { in modulated_deformable_im2col()