Home
last modified time | relevance | path

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 Ddecomposition_skip.py115 osize = decompositions.upsample_compute_output_size(
153 osize = decompositions.upsample_compute_output_size(
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py2730 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)