_test_pixel_unshuffle_error_case_helper (reference) in projects: aosp_15_r20

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)