Searched defs:full_output_size (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | UpSampleNearest1d.cpp | 26 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 D | UpSampleNearest2d.cpp | 27 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 D | UpSampleNearest3d.cpp | 30 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 D | UpSampleBilinear2d.cpp | 29 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 D | UpSampleBicubic2d.cpp | 28 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 D | UpSampleLinear1d.cpp | 28 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 D | UpSampleTrilinear3d.cpp | 30 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
|