Searched defs:h1lambda (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | UpSampleBilinear2d.cu | 58 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 D | UpSampleTrilinear3d.cu | 89 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 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 | 112 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 D | QuantizedOpKernels.cpp | 2215 const float h1lambda, 2308 const float h1lambda = h1r - h1; local
|