Home
last modified time | relevance | path

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 Dcore.py2225 def sample_inputs_elementwise_unary( function
2522 sample_inputs_func=sample_inputs_elementwise_unary,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py84 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 …]