| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | UpSampleBilinear2d.cpp | 28 float w0lambda, w1lambda; member
30 UpsampleBilinearParamW(int64_t w1, int64_t w1p, float w0lambda, float w1lambda) in UpsampleBilinearParamW()
34 , w1lambda(w1lambda) {} in UpsampleBilinearParamW()
92 const float w1lambda = w1r - w1; in upsample_bilinear2d_out_frame() local
93 const float w0lambda = static_cast<float>(1.) - w1lambda; in upsample_bilinear2d_out_frame()
95 params_w.emplace_back(w1, w1p, w0lambda, w1lambda); in upsample_bilinear2d_out_frame()
123 const float w1lambda = param_w.w1lambda; in upsample_bilinear2d_out_frame() local
127 float result = h0lambda * (w0lambda * pos1[0] + w1lambda * pos1[w1p]) + in upsample_bilinear2d_out_frame()
130 w1lambda * pos1[h1p * input_width + w1p]) - input_q_zero_point; in upsample_bilinear2d_out_frame()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | UpSampleTrilinear3d.cu | 96 const accscalar_t w1lambda = w1r - w1; in upsample_trilinear3d_out_frame() local
97 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_trilinear3d_out_frame()
104 w1lambda * idata[n][c][t1][h1][w1 + w1p]) + in upsample_trilinear3d_out_frame()
107 w1lambda * idata[n][c][t1][h1 + h1p][w1 + w1p])) + in upsample_trilinear3d_out_frame()
111 w1lambda * idata[n][c][t1 + t1p][h1][w1 + w1p]) + in upsample_trilinear3d_out_frame()
114 w1lambda * idata[n][c][t1 + t1p][h1 + h1p][w1 + w1p])); in upsample_trilinear3d_out_frame()
183 const accscalar_t w1lambda = w1r - w1; in upsample_trilinear3d_backward_out_frame() local
184 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_trilinear3d_backward_out_frame()
200 static_cast<scalar_t>(t0lambda * h0lambda * w1lambda * d2val), in upsample_trilinear3d_backward_out_frame()
212 static_cast<scalar_t>(t0lambda * h1lambda * w1lambda * d2val), in upsample_trilinear3d_backward_out_frame()
[all …]
|
| D | UpSampleLinear1d.cu | 58 const accscalar_t w1lambda = w1r - w1; in upsample_linear1d_out_frame() local
59 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_linear1d_out_frame()
64 w0lambda * idata[n][c][w1] + w1lambda * idata[n][c][w1 + w1p]; in upsample_linear1d_out_frame()
105 const accscalar_t w1lambda = w1r - w1; in upsample_linear1d_out_frame_backward() local
106 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_linear1d_out_frame_backward()
113 &idata[n][c][w1 + w1p], static_cast<scalar_t>(w1lambda * d2val)); in upsample_linear1d_out_frame_backward()
|
| D | UpSampleBilinear2d.cu | 65 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_out_frame() local
66 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_out_frame()
72 w1lambda * idata[n][c][h1][w1 + w1p]) + in upsample_bilinear2d_out_frame()
75 w1lambda * idata[n][c][h1 + h1p][w1 + w1p]); in upsample_bilinear2d_out_frame()
116 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_nhwc_out_frame() local
117 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_nhwc_out_frame()
121 w1lambda * idata[idx_cl(n, h1, w1 + w1p, c, height1, width1, channels)] in upsample_bilinear2d_nhwc_out_frame()
124 w1lambda * idata[idx_cl(n, h1 + h1p, w1 + w1p, c, height1, width1, channels)] in upsample_bilinear2d_nhwc_out_frame()
165 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_backward_out_frame() local
166 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_backward_out_frame()
[all …]
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | UpSampleMoreKernel.cpp | 467 opmath_t w0lambda, w1lambda; in cpu_upsample_linear_backward() local
472 … iw0, iw1, w0lambda, w1lambda, width_scale, ow, input_width, output_width, align_corners); in cpu_upsample_linear_backward()
475 acc_data_ptr[input_offset + iw1] += w1lambda * grad_output_value; /* i1*/ in cpu_upsample_linear_backward()
502 opmath_t h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_backward() local
510 … iw0, iw1, w0lambda, w1lambda, width_scale, ow, input_width, output_width, align_corners); in cpu_upsample_linear_backward()
513 …acc_data_ptr[input_offset + ih0 * input_width + iw1] += h0lambda * w1lambda * grad_output_value; /… in cpu_upsample_linear_backward()
515 …acc_data_ptr[input_offset + ih1 * input_width + iw1] += h1lambda * w1lambda * grad_output_value; /… in cpu_upsample_linear_backward()
545 opmath_t d0lambda, d1lambda, h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_backward() local
556 … iw0, iw1, w0lambda, w1lambda, width_scale, ow, input_width, output_width, align_corners); in cpu_upsample_linear_backward()
560 …eight * input_width + ih0 * input_width + iw1] += d0lambda * h0lambda * w1lambda * grad_output_val… in cpu_upsample_linear_backward()
[all …]
|
| D | UpSampleKernel.cpp | 610 opmath_t h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_channels_last() local
617 … iw0, iw1, w0lambda, w1lambda, width_scale, ow, input_width, output_width, align_corners); in cpu_upsample_linear_channels_last()
626 opmath_t w01 = h0lambda * w1lambda; in cpu_upsample_linear_channels_last()
628 opmath_t w11 = h1lambda * w1lambda; in cpu_upsample_linear_channels_last()
659 opmath_t d0lambda, d1lambda, h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_channels_last() local
669 … iw0, iw1, w0lambda, w1lambda, width_scale, ow, input_width, output_width, align_corners); in cpu_upsample_linear_channels_last()
683 opmath_t w001 = d0lambda * h0lambda * w1lambda; in cpu_upsample_linear_channels_last()
685 opmath_t w011 = d0lambda * h1lambda * w1lambda; in cpu_upsample_linear_channels_last()
687 opmath_t w101 = d1lambda * h0lambda * w1lambda; in cpu_upsample_linear_channels_last()
689 opmath_t w111 = d1lambda * h1lambda * w1lambda; in cpu_upsample_linear_channels_last()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 2217 const float w1lambda, argument
2248 Vectorized<float> w1lambda_v(w1lambda);
2316 const float w1lambda = w1r - w1; local
2317 const float w0lambda = static_cast<float>(1.) - w1lambda;
2340 w1lambda,
2347 (w0lambda * pos1[0] + w1lambda * pos1[w1p * channels]) +
2350 w1lambda * pos1[(h1p * input_width + w1p) * channels]);
|