/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
H A D | UpSampleKernel.cpp | 1703 std::optional<double> scales_h, in upsample_nearest2d_kernel_impl() 1719 std::optional<double> scales_h, in _upsample_nearest_exact2d_kernel_impl() 1735 std::optional<double> scales_h, in upsample_nearest3d_kernel_impl() 1752 std::optional<double> scales_h, in _upsample_nearest_exact3d_kernel_impl() 1778 std::optional<double> scales_h, in upsample_bilinear2d_kernel_impl_float() 1800 std::optional<double> scales_h, in upsample_bilinear2d_kernel_impl() 1829 std::optional<double> scales_h, in upsample_bilinear2d_aa_kernel_impl() 1853 std::optional<double> scales_h, in upsample_trilinear3d_kernel_impl() 1869 std::optional<double> scales_h, in upsample_bicubic2d_kernel_impl() 1899 std::optional<double> scales_h, in upsample_bicubic2d_aa_kernel_impl() [all …]
|
H A D | UpSampleMoreKernel.cpp | 358 std::optional<double> scales_h, in upsample_nearest2d_backward_kernel_impl() 374 std::optional<double> scales_h, in _upsample_nearest_exact2d_backward_kernel_impl() 391 std::optional<double> scales_h, in upsample_nearest3d_backward_kernel_impl() 408 std::optional<double> scales_h, in _upsample_nearest_exact3d_backward_kernel_impl() 758 std::optional<double> scales_h, in upsample_bilinear2d_backward_kernel_impl() 776 std::optional<double> scales_h, in upsample_trilinear3d_backward_kernel_impl()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
H A D | UpSampleNearest2d.cpp | 39 std::optional<double> scales_h, in upsample_nearest2d_out_frame() 95 std::optional<double> scales_h, in upsample_nearest2d_out_frame_nhwc() 124 std::optional<double> scales_h, in _upsample_nearest2d_quantized_cpu()
|
H A D | UpSampleNearest3d.cpp | 40 std::optional<double> scales_h, in upsample_nearest3d_out_frame() 97 std::optional<double> scales_h, in upsample_nearest3d_out_frame_nhwc() 137 std::optional<double> scales_h, in _upsample_nearest3d_quantized_cpu()
|
H A D | UpSampleBilinear2d.cpp | 49 std::optional<double> scales_h, in upsample_bilinear2d_out_frame() 149 std::optional<double> scales_h, in upsample_bilinear2d_quantized_cpu()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
H A D | Upsample.cpp | 15 const std::optional<double> scales_h, in upsample_nearest2d() 101 const std::optional<double> scales_h, in upsample_bilinear2d()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | UpSampleBilinear2d.cu | 267 std::optional<double> scales_h, in upsample_bilinear2d_out_cuda_template() 365 std::optional<double> scales_h, in upsample_bilinear2d_backward_out_cuda_template() 679 std::optional<double> scales_h, in upsample_gen2d_aa_out_cuda_template() 775 std::optional<double> scales_h, in upsample_gen2d_aa_backward_out_cuda_template()
|
H A D | UpSampleBicubic2d.cu | 173 std::optional<double> scales_h, in upsample_bicubic2d_out_cuda_template() 228 std::optional<double> scales_h, in upsample_bicubic2d_backward_out_cuda_template()
|
H A D | UpSampleNearest3d.cu | 152 std::optional<double> scales_h, in upsample_nearest3d_out_cuda_template() 228 std::optional<double> scales_h, in upsample_nearest3d_backward_out_cuda_template()
|
H A D | UpSampleTrilinear3d.cu | 249 std::optional<double> scales_h, in upsample_trilinear3d_out_cuda_template() 305 std::optional<double> scales_h, in upsample_trilinear3d_backward_out_cuda_template()
|
H A D | UpSampleNearest2d.cu | 210 std::optional<double> scales_h, in upsample_nearest2d_out_cuda_template() 340 std::optional<double> scales_h, in upsample_nearest2d_backward_out_cuda_template()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | UpSampleBicubic2d.cpp | 118 std::optional<double> scales_h, in upsample_bicubic2d_backward_out_frame() 188 std::optional<double> scales_h, in upsample_bicubic2d_backward_kernel()
|
/aosp_15_r20/external/pytorch/torch/onnx/ |
H A D | symbolic_caffe2.py | 210 scales_h=None, argument
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _meta_registrations.py | 5761 def upsample_nearest2d(input, output_size, scales_h=None, scales_w=None): argument 5822 def upsample_nearest3d(input, output_size, scales_d=None, scales_h=None, scales_w=None): argument 6233 scales_h=None, argument
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | lowering.py | 4482 x, output_size=None, input_size=None, scales_h=None, scales_w=None argument
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
H A D | QuantizedOpKernels.cpp | 2280 std::optional<double> scales_h,
|