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 …]
|