Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleBilinear2d.cu58 const accscalar_t h1lambda = h1r - h1; in upsample_bilinear2d_out_frame() local
109 const accscalar_t h1lambda = h1r - h1; in upsample_bilinear2d_nhwc_out_frame() local
158 const accscalar_t h1lambda = h1r - h1; in upsample_bilinear2d_backward_out_frame() local
224 const accscalar_t h1lambda = h1r - h1; in upsample_bilinear2d_backward_nhwc_out_frame() local
H A DUpSampleTrilinear3d.cu89 const accscalar_t h1lambda = h1r - h1; in upsample_trilinear3d_out_frame() local
176 const accscalar_t h1lambda = h1r - h1; 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.cpp112 const float h1lambda = h1r - h1; in upsample_bilinear2d_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp2215 const float h1lambda,
2308 const float h1lambda = h1r - h1; local