Searched defs:sample_args (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_ops_jit.py | 355 def _fn(*sample_args, **sample_kwargs):
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/ |
| H A D | reinplace.py | 252 def reinplace(gm, *sample_args):
|
| /aosp_15_r20/external/pytorch/torch/cuda/ |
| H A D | graphs.py | 192 callables, sample_args, num_warmup_iters=3, allow_unused_input=False, pool=None argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 161 f, args, kwargs, sample_args, output_process_fn_grad=None argument
|