Searched refs:upsample_compute_output_size (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | decomposition_skip.py | 115 osize = decompositions.upsample_compute_output_size(
153 osize = decompositions.upsample_compute_output_size(
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2730 def upsample_compute_output_size(input_size, output_size, scale_factors): function
2778 osize = upsample_compute_output_size(input.size(), output_size, scale_factors)
2799 osize = upsample_compute_output_size(input.size(), output_size, scale_factors)
3607 osize = upsample_compute_output_size(input.size(), output_size, scale_factors)
3619 osize = upsample_compute_output_size(input.size(), output_size, scale_factors)
3636 osize = upsample_compute_output_size(input.size(), output_size, scale_factors)
|