Searched refs:sample_inputs_split (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 6991 def sample_inputs_split(op_info, device, dtype, requires_grad, *, list_args=False, **kwargs): function
17201 sample_inputs_func=partial(sample_inputs_split, list_args=False),
17214 sample_inputs_func=partial(sample_inputs_split, list_args=True),
17221 sample_inputs_func=partial(sample_inputs_split, list_args=False),
|