Searched defs:allow_unused_input (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/cuda/ |
| D | graphs.py | 192 callables, sample_args, num_warmup_iters=3, allow_unused_input=False, pool=None argument
|
| /external/pytorch/test/ |
| D | test_cuda.py | 2628 self, with_amp, cache_enabled, allow_unused_input argument
2779 self, with_amp, cache_enabled, allow_unused_input argument
|