Searched defs:sample_inputs_func (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 691 sample_inputs_func: Callable = None variable in OpInfo
1605 def sample_inputs_func(*args, **kwargs): function
2169 sample_inputs_func=sample_inputs_elementwise_binary, argument
2522 sample_inputs_func=sample_inputs_elementwise_unary, argument
2666 sample_inputs_func=sample_inputs_spectral_ops, argument
2706 sample_inputs_func=None, argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 6364 def reference_inputs_elementwise_ternary(op, device, dtype, requires_grad, *, sample_inputs_func, s… argument
|