Searched defs:use_device (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/autograd/ | ||
| D | profiler.py | 202 use_device=None, argument |
| D | profiler_util.py | 457 use_device=None, argument |
| /external/pytorch/test/jit/ | ||
| D | test_tracer.py | 1351 def use_device(x): function |