Searched refs:compute_scales_value (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | UpSampleNearest3d.cpp | 42 float depth_scale = compute_scales_value<float>(scales_d, input_depth, output_depth); in upsample_nearest3d_out_frame()
43 float height_scale = compute_scales_value<float>(scales_h, input_height, output_height); in upsample_nearest3d_out_frame()
44 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest3d_out_frame()
99 float depth_scale = compute_scales_value<float>(scales_d, input_depth, output_depth); in upsample_nearest3d_out_frame_nhwc()
100 float height_scale = compute_scales_value<float>(scales_h, input_height, output_height); in upsample_nearest3d_out_frame_nhwc()
101 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest3d_out_frame_nhwc()
|
| H A D | UpSampleNearest2d.cpp | 41 float height_scale = compute_scales_value<float>(scales_h, input_height, output_height); in upsample_nearest2d_out_frame()
42 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest2d_out_frame()
97 float height_scale = compute_scales_value<float>(scales_h, input_height, output_height); in upsample_nearest2d_out_frame_nhwc()
98 float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest2d_out_frame_nhwc()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Upsample.cpp | 59 compute_scales_value<float>( in upsample_nearest2d()
63 compute_scales_value<float>( in upsample_nearest2d()
137 compute_scales_value<float>( in upsample_bilinear2d()
141 compute_scales_value<float>( in upsample_bilinear2d()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | UpSample.h | 254 inline scalar_t compute_scales_value( in compute_scales_value() function
279 return compute_scales_value<scalar_t>(scale, input_size, output_size); in area_pixel_compute_scale()
350 float scale = compute_scales_value<float>(scales, input_size, output_size); in nearest_idx()
360 float scale = compute_scales_value<float>(scales, input_size, output_size); in nearest_exact_idx()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest3d.cu | 194 const float depth_scale = compute_scales_value<float>(scales_d, input_depth, output_depth); in upsample_nearest3d_out_cuda_template()
195 … const float height_scale = compute_scales_value<float>(scales_h, input_height, output_height); in upsample_nearest3d_out_cuda_template()
196 const float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest3d_out_cuda_template()
|
| H A D | UpSample.cuh | 75 __host__ __forceinline__ accscalar_t compute_scales_value( in compute_scales_value() function
110 return compute_scales_value<accscalar_t>(scale, input_size, output_size); in area_pixel_compute_scale()
|
| H A D | UpSampleNearest2d.cu | 227 const float height_scale = compute_scales_value<float>(scales_h, input_height, output_height); in upsample_nearest2d_out_cuda_template()
228 const float width_scale = compute_scales_value<float>(scales_w, input_width, output_width); in upsample_nearest2d_out_cuda_template()
|
| H A D | UpSampleNearest1d.cu | 138 const float scale_factor = compute_scales_value<float>(scales, input_width, output_width); in upsample_nearest1d_out_cuda_template()
|