Searched defs:h0lambda (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | UpSampleBilinear2d.cu | 59 const accscalar_t h0lambda = static_cast<accscalar_t>(1) - h1lambda; in upsample_bilinear2d_out_frame() local 110 const accscalar_t h0lambda = static_cast<accscalar_t>(1) - h1lambda; in upsample_bilinear2d_nhwc_out_frame() local 159 const accscalar_t h0lambda = static_cast<accscalar_t>(1) - h1lambda; in upsample_bilinear2d_backward_out_frame() local 225 const accscalar_t h0lambda = static_cast<accscalar_t>(1) - h1lambda; in upsample_bilinear2d_backward_nhwc_out_frame() local
|
H A D | UpSampleTrilinear3d.cu | 90 const accscalar_t h0lambda = static_cast<accscalar_t>(1) - h1lambda; in upsample_trilinear3d_out_frame() local 177 const accscalar_t h0lambda = static_cast<accscalar_t>(1) - h1lambda; in upsample_trilinear3d_backward_out_frame() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
H A D | UpSampleMoreKernel.cpp | 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/quantized/cpu/ |
H A D | UpSampleBilinear2d.cpp | 113 const float h0lambda = static_cast<float>(1.) - h1lambda; in upsample_bilinear2d_out_frame() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
H A D | QuantizedOpKernels.cpp | 2214 const float h0lambda, 2309 const float h0lambda = static_cast<float>(1.) - h1lambda; local
|