Searched refs:compute_scales_value_backwards (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest3d.cu | 270 … float depth_scale = compute_scales_value_backwards<float>(scales_d, output_depth, input_depth); in upsample_nearest3d_backward_out_cuda_template()
271 … float height_scale = compute_scales_value_backwards<float>(scales_h, output_height, input_height); in upsample_nearest3d_backward_out_cuda_template()
272 … float width_scale = compute_scales_value_backwards<float>(scales_w, output_width, input_width); in upsample_nearest3d_backward_out_cuda_template()
|
| H A D | UpSampleNearest2d.cu | 358 …const float height_scale = compute_scales_value_backwards<float>(scales_h, output_height, input_he… in upsample_nearest2d_backward_out_cuda_template()
359 …const float width_scale = compute_scales_value_backwards<float>(scales_w, output_width, input_widt… in upsample_nearest2d_backward_out_cuda_template()
|
| H A D | UpSampleNearest1d.cu | 189 …const float scale_factor = compute_scales_value_backwards<float>(scales, output_width, input_width… in upsample_nearest1d_backward_out_cuda_template()
|
| H A D | UpSample.cuh | 86 __host__ __forceinline__ accscalar_t compute_scales_value_backwards( in compute_scales_value_backwards() function
|