Home
last modified time | relevance | path

Searched defs:scales_h (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleKernel.cpp1703 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 DUpSampleMoreKernel.cpp358 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 DUpSampleNearest2d.cpp39 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 DUpSampleNearest3d.cpp40 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 DUpSampleBilinear2d.cpp49 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 DUpsample.cpp15 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 DUpSampleBilinear2d.cu267 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 DUpSampleBicubic2d.cu173 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 DUpSampleNearest3d.cu152 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 DUpSampleTrilinear3d.cu249 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 DUpSampleNearest2d.cu210 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 DUpSampleBicubic2d.cpp118 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 Dsymbolic_caffe2.py210 scales_h=None, argument
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py5761 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 Dlowering.py4482 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 DQuantizedOpKernels.cpp2280 std::optional<double> scales_h,