Searched refs:generate_input_from_mask (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 4429 def generate_input_from_mask(mask_shape, dim): function
4437 generate_input_from_mask((2, 3, 4), dim=1),
4439 generate_input_from_mask((5, 5), dim=0),
4441 generate_input_from_mask((2, 3, 4, 5), dim=2),
4443 generate_input_from_mask((10,), dim=0),
4445 generate_input_from_mask((1, 5, 5), dim=1),
|