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),
|