Searched defs:w0lambda (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | UpSampleBilinear2d.cpp | 28 float w0lambda, w1lambda; member
93 const float w0lambda = static_cast<float>(1.) - w1lambda; in upsample_bilinear2d_out_frame() local
122 const float w0lambda = param_w.w0lambda; in upsample_bilinear2d_out_frame() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | UpSampleMoreKernel.cpp | 467 opmath_t w0lambda, w1lambda; in cpu_upsample_linear_backward() local
502 opmath_t h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_backward() local
545 opmath_t d0lambda, d1lambda, h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_backward() local
649 opmath_t h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_backward_channels_last() local
698 opmath_t d0lambda, d1lambda, h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_backward_channels_last() local
|
| H A D | UpSampleKernel.cpp | 610 opmath_t h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_channels_last() local
659 opmath_t d0lambda, d1lambda, h0lambda, h1lambda, w0lambda, w1lambda; in cpu_upsample_linear_channels_last() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleBilinear2d.cu | 66 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_out_frame() local
117 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_nhwc_out_frame() local
166 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_backward_out_frame() local
232 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_bilinear2d_backward_nhwc_out_frame() local
|
| H A D | UpSampleLinear1d.cu | 59 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_linear1d_out_frame() local
106 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_linear1d_out_frame_backward() local
|
| H A D | UpSampleTrilinear3d.cu | 97 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_trilinear3d_out_frame() local
184 const accscalar_t w0lambda = static_cast<accscalar_t>(1) - w1lambda; in upsample_trilinear3d_backward_out_frame() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 2216 const float w0lambda,
2317 const float w0lambda = static_cast<float>(1.) - w1lambda; local
|