Searched refs:sample_inputs_elementwise_unary (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 2225 def sample_inputs_elementwise_unary( function
2522 sample_inputs_func=sample_inputs_elementwise_unary,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 84 sample_inputs_elementwise_unary,
542 yield from sample_inputs_elementwise_unary(op_info, device, dtype, requires_grad,
2916 yield from sample_inputs_elementwise_unary(
6260 yield from sample_inputs_elementwise_unary(op_info, device, dtype, requires_grad)
6271 yield from sample_inputs_elementwise_unary(op_info, device, dtype, requires_grad)
6282 yield from sample_inputs_elementwise_unary(op_info, device, dtype, requires_grad)
17050 … sample_inputs_func=partial(sample_inputs_elementwise_unary, op_kwargs={'decimals': 0}),
17062 … sample_inputs_func=partial(sample_inputs_elementwise_unary, op_kwargs={'decimals': 3}),
17088 … sample_inputs_func=partial(sample_inputs_elementwise_unary, op_kwargs={'decimals': -3}),
19438 sample_inputs_func=partial(sample_inputs_elementwise_unary, op_kwargs={'value': True}),
[all …]
|