Searched refs:_test_pixel_unshuffle_error_case_helper (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_mps.py | 1354 …def _test_pixel_unshuffle_error_case_helper(num_input_dims, valid_height_dim=True, valid_width_dim… function 1393 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, valid_height_dim=False) 1394 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, valid_width_dim=False) 1395 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, downscale_factor=0) 1396 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, downscale_factor=-2)
|
H A D | test_nn.py | 4442 …def _test_pixel_unshuffle_error_case_helper(num_input_dims, valid_height_dim=True, valid_width_dim… function 4473 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, valid_height_dim=False) 4474 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, valid_width_dim=False) 4475 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, downscale_factor=0) 4476 … _test_pixel_unshuffle_error_case_helper(num_input_dims=num_input_dims, downscale_factor=-2)
|