Searched refs:sample_inputs_max_pool (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 3661 def sample_inputs_max_pool(op_info, device, dtype, requires_grad, **kwargs): function
9163 for sample in sample_inputs_max_pool(pool_op_info, device, dtype, requires_grad, **kwargs):
15658 sample_inputs_func=sample_inputs_max_pool),
15674 sample_inputs_func=sample_inputs_max_pool),
15690 sample_inputs_func=sample_inputs_max_pool,
15716 sample_inputs_func=sample_inputs_max_pool),
|