Home
last modified time | relevance | path

Searched refs:_test_pixel_unshuffle_error_case_helper (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_mps.py1354 …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 Dtest_nn.py4442 …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)