Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
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
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
/external/pytorch/aten/src/ATen/native/cpu/
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
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
/external/pytorch/aten/src/ATen/native/quantized/cpu/
DUpSampleBilinear2d.cpp112 const float h1lambda = h1r - h1; in upsample_bilinear2d_out_frame() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp2215 const float h1lambda,
2308 const float h1lambda = h1r - h1; local