argmax_w_low (reference) in projects: dports

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:argmax_w_low (Results 1 – 6 of 6) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A D	deformable_im2col.h	134 if (argmax_w_low >= width - 1) { in get_gradient_weight_cpu() 135 argmax_w_high = argmax_w_low = width - 1; in get_gradient_weight_cpu() 136 argmax_w = static_cast<DType>(argmax_w_low); in get_gradient_weight_cpu() 138 argmax_w_high = argmax_w_low + 1; in get_gradient_weight_cpu() 142 if (w == argmax_w_low) { in get_gradient_weight_cpu() 148 if (w == argmax_w_low) { in get_gradient_weight_cpu() 181 if (argmax_w_low >= width - 1) { in get_coordinate_weight_cpu() 182 argmax_w_high = argmax_w_low = width - 1; in get_coordinate_weight_cpu() 183 argmax_w = static_cast<DType>(argmax_w_low); in get_coordinate_weight_cpu() 185 argmax_w_high = argmax_w_low + 1; in get_coordinate_weight_cpu() [all …]
H A D	deformable_im2col.cuh	138 if (argmax_w_low >= width - 1) in get_gradient_weight() 140 argmax_w_high = argmax_w_low = width - 1; in get_gradient_weight() 141 argmax_w = static_cast<DType>(argmax_w_low); in get_gradient_weight() 143 argmax_w_high = argmax_w_low + 1; in get_gradient_weight() 147 if (w == argmax_w_low) { in get_gradient_weight() 153 if (w == argmax_w_low) { in get_gradient_weight() 188 if (argmax_w_low >= width - 1) { in get_coordinate_weight() 189 argmax_w_high = argmax_w_low = width - 1; in get_coordinate_weight() 190 argmax_w = static_cast<DType>(argmax_w_low); in get_coordinate_weight() 192 argmax_w_high = argmax_w_low + 1; in get_coordinate_weight() [all …]
H A D	modulated_deformable_im2col.cuh	120 int argmax_w_low = floor(argmax_w); in dmcn_get_gradient_weight() local 122 int argmax_w_high = argmax_w_low + 1; in dmcn_get_gradient_weight() 125 if (h == argmax_h_low && w == argmax_w_low) in dmcn_get_gradient_weight() 129 if (h == argmax_h_high && w == argmax_w_low) in dmcn_get_gradient_weight() 149 int argmax_w_low = floor(argmax_w); in dmcn_get_coordinate_weight() local 151 int argmax_w_high = argmax_w_low + 1; in dmcn_get_coordinate_weight() 156 if (argmax_h_low >= 0 && argmax_w_low >= 0) in dmcn_get_coordinate_weight() 157 … weight += -1 * (argmax_w_low + 1 - argmax_w) * im_data[argmax_h_low * data_width + argmax_w_low]; in dmcn_get_coordinate_weight() 160 if (argmax_h_high <= height - 1 && argmax_w_low >= 0) in dmcn_get_coordinate_weight() 161 … weight += (argmax_w_low + 1 - argmax_w) * im_data[argmax_h_high * data_width + argmax_w_low]; in dmcn_get_coordinate_weight() [all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/nn/
H A D	deformable_im2col.h	134 if (argmax_w_low >= width - 1) { in get_gradient_weight_cpu() 135 argmax_w_high = argmax_w_low = width - 1; in get_gradient_weight_cpu() 136 argmax_w = static_cast<DType>(argmax_w_low); in get_gradient_weight_cpu() 138 argmax_w_high = argmax_w_low + 1; in get_gradient_weight_cpu() 142 if (w == argmax_w_low) { in get_gradient_weight_cpu() 148 if (w == argmax_w_low) { in get_gradient_weight_cpu() 181 if (argmax_w_low >= width - 1) { in get_coordinate_weight_cpu() 182 argmax_w_high = argmax_w_low = width - 1; in get_coordinate_weight_cpu() 183 argmax_w = static_cast<DType>(argmax_w_low); in get_coordinate_weight_cpu() 185 argmax_w_high = argmax_w_low + 1; in get_coordinate_weight_cpu() [all …]
H A D	deformable_im2col.cuh	138 if (argmax_w_low >= width - 1) in get_gradient_weight() 140 argmax_w_high = argmax_w_low = width - 1; in get_gradient_weight() 141 argmax_w = static_cast<DType>(argmax_w_low); in get_gradient_weight() 143 argmax_w_high = argmax_w_low + 1; in get_gradient_weight() 147 if (w == argmax_w_low) { in get_gradient_weight() 153 if (w == argmax_w_low) { in get_gradient_weight() 188 if (argmax_w_low >= width - 1) { in get_coordinate_weight() 189 argmax_w_high = argmax_w_low = width - 1; in get_coordinate_weight() 190 argmax_w = static_cast<DType>(argmax_w_low); in get_coordinate_weight() 192 argmax_w_high = argmax_w_low + 1; in get_coordinate_weight() [all …]
H A D	modulated_deformable_im2col.cuh	120 int argmax_w_low = floor(argmax_w); in dmcn_get_gradient_weight() local 122 int argmax_w_high = argmax_w_low + 1; in dmcn_get_gradient_weight() 125 if (h == argmax_h_low && w == argmax_w_low) in dmcn_get_gradient_weight() 129 if (h == argmax_h_high && w == argmax_w_low) in dmcn_get_gradient_weight() 149 int argmax_w_low = floor(argmax_w); in dmcn_get_coordinate_weight() local 151 int argmax_w_high = argmax_w_low + 1; in dmcn_get_coordinate_weight() 156 if (argmax_h_low >= 0 && argmax_w_low >= 0) in dmcn_get_coordinate_weight() 157 … weight += -1 * (argmax_w_low + 1 - argmax_w) * im_data[argmax_h_low * data_width + argmax_w_low]; in dmcn_get_coordinate_weight() 160 if (argmax_h_high <= height - 1 && argmax_w_low >= 0) in dmcn_get_coordinate_weight() 161 … weight += (argmax_w_low + 1 - argmax_w) * im_data[argmax_h_high * data_width + argmax_w_low]; in dmcn_get_coordinate_weight() [all …]