Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py4429 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),