Home
last modified time | relevance | path

Searched defs:w1lambda (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleBilinear2d.cpp28 float w0lambda, w1lambda; member
92 const float w1lambda = w1r - w1; in upsample_bilinear2d_out_frame() local
123 const float w1lambda = param_w.w1lambda; in upsample_bilinear2d_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp467 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 DUpSampleKernel.cpp610 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 DUpSampleBilinear2d.cu65 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_out_frame() local
116 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_nhwc_out_frame() local
165 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_backward_out_frame() local
231 const accscalar_t w1lambda = w1r - w1; in upsample_bilinear2d_backward_nhwc_out_frame() local
H A DUpSampleLinear1d.cu58 const accscalar_t w1lambda = w1r - w1; in upsample_linear1d_out_frame() local
105 const accscalar_t w1lambda = w1r - w1; in upsample_linear1d_out_frame_backward() local
H A DUpSampleTrilinear3d.cu96 const accscalar_t w1lambda = w1r - w1; in upsample_trilinear3d_out_frame() local
183 const accscalar_t w1lambda = w1r - w1; in upsample_trilinear3d_backward_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp2217 const float w1lambda,
2316 const float w1lambda = w1r - w1; local