| /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()
33 , w0lambda(w0lambda) in UpsampleBilinearParamW()
93 const float w0lambda = static_cast<float>(1.) - w1lambda; in upsample_bilinear2d_out_frame() local
95 params_w.emplace_back(w1, w1p, w0lambda, w1lambda); in upsample_bilinear2d_out_frame()
122 const float w0lambda = param_w.w0lambda; in upsample_bilinear2d_out_frame() local
127 float result = h0lambda * (w0lambda * pos1[0] + w1lambda * pos1[w1p]) + in upsample_bilinear2d_out_frame()
129 (w0lambda * pos1[h1p * input_width] + in upsample_bilinear2d_out_frame()
|
| /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()
474 acc_data_ptr[input_offset + iw0] += w0lambda * grad_output_value; /* i0 */ 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()
512 …acc_data_ptr[input_offset + ih0 * input_width + iw0] += h0lambda * w0lambda * grad_output_value; /… in cpu_upsample_linear_backward()
514 …acc_data_ptr[input_offset + ih1 * input_width + iw0] += h1lambda * w0lambda * 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()
559 …eight * input_width + ih0 * input_width + iw0] += d0lambda * h0lambda * w0lambda * 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()
625 opmath_t w00 = h0lambda * w0lambda; in cpu_upsample_linear_channels_last()
627 opmath_t w10 = h1lambda * w0lambda; 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()
682 opmath_t w000 = d0lambda * h0lambda * w0lambda; in cpu_upsample_linear_channels_last()
684 opmath_t w010 = d0lambda * h1lambda * w0lambda; in cpu_upsample_linear_channels_last()
686 opmath_t w100 = d1lambda * h0lambda * w0lambda; in cpu_upsample_linear_channels_last()
688 opmath_t w110 = d1lambda * h1lambda * w0lambda; in cpu_upsample_linear_channels_last()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | UpSampleTrilinear3d.cu | 97 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_trilinear3d_out_frame() local
103 (w0lambda * idata[n][c][t1][h1][w1] + in upsample_trilinear3d_out_frame()
106 (w0lambda * idata[n][c][t1][h1 + h1p][w1] + in upsample_trilinear3d_out_frame()
110 (w0lambda * idata[n][c][t1 + t1p][h1][w1] + in upsample_trilinear3d_out_frame()
113 (w0lambda * idata[n][c][t1 + t1p][h1 + h1p][w1] + in upsample_trilinear3d_out_frame()
184 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_trilinear3d_backward_out_frame() local
194 static_cast<scalar_t>(t0lambda * h0lambda * w0lambda * d2val), in upsample_trilinear3d_backward_out_frame()
206 static_cast<scalar_t>(t0lambda * h1lambda * w0lambda * d2val), in upsample_trilinear3d_backward_out_frame()
218 static_cast<scalar_t>(t1lambda * h0lambda * w0lambda * d2val), in upsample_trilinear3d_backward_out_frame()
230 static_cast<scalar_t>(t1lambda * h1lambda * w0lambda * d2val), in upsample_trilinear3d_backward_out_frame()
|
| D | UpSampleLinear1d.cu | 59 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_linear1d_out_frame() local
64 w0lambda * idata[n][c][w1] + w1lambda * idata[n][c][w1 + w1p]; in upsample_linear1d_out_frame()
106 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_linear1d_out_frame_backward() local
111 gpuAtomicAddNoReturn(&idata[n][c][w1], static_cast<scalar_t>(w0lambda * d2val)); in upsample_linear1d_out_frame_backward()
|
| D | UpSampleBilinear2d.cu | 66 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_out_frame() local
71 (w0lambda * idata[n][c][h1][w1] + in upsample_bilinear2d_out_frame()
74 (w0lambda * idata[n][c][h1 + h1p][w1] + in upsample_bilinear2d_out_frame()
117 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_nhwc_out_frame() local
120 w0lambda * idata[idx_cl(n, h1, w1, c, height1, width1, channels)] + in upsample_bilinear2d_nhwc_out_frame()
123 w0lambda * idata[idx_cl(n, h1 + h1p, w1, c, height1, width1, channels)] + in upsample_bilinear2d_nhwc_out_frame()
166 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_backward_out_frame() local
173 static_cast<scalar_t>(h0lambda * w0lambda * d2val), in upsample_bilinear2d_backward_out_frame()
185 static_cast<scalar_t>(h1lambda * w0lambda * d2val), in upsample_bilinear2d_backward_out_frame()
232 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_backward_nhwc_out_frame() local
[all …]
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 2216 const float w0lambda, argument
2247 Vectorized<float> w0lambda_v(w0lambda);
2317 const float w0lambda = static_cast<float>(1.) - w1lambda; local
2339 w0lambda,
2347 (w0lambda * pos1[0] + w1lambda * pos1[w1p * channels]) +
2349 (w0lambda * pos1[h1p * input_width * channels] +
|