Home
last modified time | relevance | path

Searched defs:h0lambda (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleBilinear2d.cu59 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 DUpSampleTrilinear3d.cu90 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 DUpSampleMoreKernel.cpp502 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/quantized/cpu/
H A DUpSampleBilinear2d.cpp113 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 DQuantizedOpKernels.cpp2214 const float h0lambda,
2309 const float h0lambda = static_cast<float>(1.) - h1lambda; local