Searched refs:sample_inputs_jiterator (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 714 def sample_inputs_jiterator(op, device, dtype, requires_grad, **kwargs): function
20109 sample_inputs_func=partial(sample_inputs_jiterator, num_inputs=2, alpha=-3.14),
20135 sample_inputs_func=partial(sample_inputs_jiterator, num_inputs=4, alpha=3.14, beta=-4.20),
20165 sample_inputs_func=partial(sample_inputs_jiterator, num_inputs=2, alpha=-0.42),
20197 sample_inputs_func=partial(sample_inputs_jiterator, num_inputs=2),
|