Home
last modified time | relevance | path

Searched defs:full_output_size (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleNearest1d.cpp26 auto full_output_size = native::upsample_1d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
40 auto full_output_size = native::upsample_1d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
54 auto full_output_size = native::upsample_1d_common_check(input_size, output_size); in TORCH_META_FUNC() local
66 auto full_output_size = native::upsample_1d_common_check(input_size, output_size); in TORCH_META_FUNC() local
H A DUpSampleNearest2d.cpp27 auto full_output_size = native::upsample_2d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
41 auto full_output_size = native::upsample_2d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
59 auto full_output_size = native::upsample_2d_common_check(input_size, output_size); in TORCH_META_FUNC() local
83 auto full_output_size = native::upsample_2d_common_check(input_size, output_size); in TORCH_META_FUNC() local
H A DUpSampleNearest3d.cpp30 auto full_output_size = native::upsample_3d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
48 auto full_output_size = native::upsample_3d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
67 auto full_output_size = native::upsample_3d_common_check(input_size, output_size); in TORCH_META_FUNC() local
92 auto full_output_size = native::upsample_3d_common_check(input_size, output_size); in TORCH_META_FUNC() local
H A DUpSampleBilinear2d.cpp29 auto full_output_size = native::upsample_2d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
48 auto full_output_size = native::upsample_2d_common_check(input_size, output_size); in TORCH_META_FUNC() local
68 auto full_output_size = native::upsample_2d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
87 auto full_output_size = native::upsample_2d_common_check(input_size, output_size); in TORCH_META_FUNC() local
H A DUpSampleBicubic2d.cpp28 auto full_output_size = native::upsample_2d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
47 auto full_output_size = native::upsample_2d_common_check(input_size, output_size); in TORCH_META_FUNC() local
67 auto full_output_size = native::upsample_2d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
86 auto full_output_size = native::upsample_2d_common_check(input_size, output_size); in TORCH_META_FUNC() local
H A DUpSampleLinear1d.cpp28 auto full_output_size = native::upsample_1d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
46 auto full_output_size = native::upsample_1d_common_check(input_size, output_size); in TORCH_META_FUNC() local
H A DUpSampleTrilinear3d.cpp30 auto full_output_size = native::upsample_3d_common_check(input.sizes(), output_size); in TORCH_META_FUNC() local
50 auto full_output_size = native::upsample_3d_common_check(input_size, output_size); in TORCH_META_FUNC() local