Searched refs:_test_interpolate_helper (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 6543 def _test_interpolate_helper(in_t, scale_factor, layer): function
6573 _test_interpolate_helper(input, scale_factor, m)
6578 _test_interpolate_helper(_make_input(1, device), scale_factor, m)
6582 _test_interpolate_helper(_make_input(2, device), scale_factor, m)
6588 _test_interpolate_helper(_make_input(2, device), scale_factor, m)
6592 _test_interpolate_helper(_make_input(3, device), scale_factor, m)
|