Home
last modified time | relevance | path

Searched refs:area_pixel_compute_source_index (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleTrilinear3d.cu78 const accscalar_t t1r = area_pixel_compute_source_index<accscalar_t>( in upsample_trilinear3d_out_frame()
85 const accscalar_t h1r = area_pixel_compute_source_index<accscalar_t>( in upsample_trilinear3d_out_frame()
92 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_trilinear3d_out_frame()
165 const accscalar_t t1r = area_pixel_compute_source_index<accscalar_t>( in upsample_trilinear3d_backward_out_frame()
172 const accscalar_t h1r = area_pixel_compute_source_index<accscalar_t>( in upsample_trilinear3d_backward_out_frame()
179 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_trilinear3d_backward_out_frame()
H A DUpSampleBicubic2d.cu59 accscalar_t real_x = area_pixel_compute_source_index( in upsample_bicubic2d_out_frame()
64 accscalar_t real_y = area_pixel_compute_source_index( in upsample_bicubic2d_out_frame()
132 accscalar_t real_x = area_pixel_compute_source_index( in upsample_bicubic2d_backward_out_frame()
137 accscalar_t real_y = area_pixel_compute_source_index( in upsample_bicubic2d_backward_out_frame()
H A DUpSampleBilinear2d.cu54 const accscalar_t h1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_out_frame()
61 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_out_frame()
105 const accscalar_t h1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_nhwc_out_frame()
112 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_nhwc_out_frame()
154 const accscalar_t h1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_backward_out_frame()
161 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_backward_out_frame()
220 const accscalar_t h1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_backward_nhwc_out_frame()
227 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_bilinear2d_backward_nhwc_out_frame()
H A DUpSampleLinear1d.cu54 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_linear1d_out_frame()
101 const accscalar_t w1r = area_pixel_compute_source_index<accscalar_t>( in upsample_linear1d_out_frame_backward()
H A DUpSample.cuh115 __device__ __forceinline__ accscalar_t area_pixel_compute_source_index( in area_pixel_compute_source_index() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleBilinear2d.cpp86 const auto w1r = area_pixel_compute_source_index<float>( in upsample_bilinear2d_out_frame()
106 const auto h1r = area_pixel_compute_source_index<float>( in upsample_bilinear2d_out_frame()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp143 …const opmath_t real_x = area_pixel_compute_source_index(width_scale, output_x, align_corners, /*cu… in upsample_bicubic2d_backward_out_frame()
148 …const opmath_t real_y = area_pixel_compute_source_index(height_scale, output_y, align_corners, /*c… in upsample_bicubic2d_backward_out_frame()
H A DUpSample.h284 inline scalar_t area_pixel_compute_source_index( in area_pixel_compute_source_index() function
464 area_pixel_compute_source_index<opmath_t>( in compute_source_index_and_lambda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleKernel.cpp802 const auto real_input_index = area_pixel_compute_source_index<scalar_t>( in _compute_indices_min_size_weights()
1093 area_pixel_compute_source_index<opmath_t>( in compute_indices_weights()
1143 area_pixel_compute_source_index<opmath_t>( in compute_indices_weights()
1311 area_pixel_compute_source_index<opmath_t>( in compute_indices_weights()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp2303 const auto h1r = area_pixel_compute_source_index<float>(
2311 const auto w1r = area_pixel_compute_source_index<float>(