Searched refs:_run_py_multithread_fn (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 12689 def _run_py_multithread_fn( member in TestMultithreadAutograd
12728 self._run_py_multithread_fn(fn)
12739 self._run_py_multithread_fn(train_fn)
12749 self._run_py_multithread_fn(train_fn_backward, (x,))
12765 self._run_py_multithread_fn(train_fn_grad, (x,))
12797 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5)
12833 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5)
12870 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5)
12905 self._run_py_multithread_fn(backward_retain_graph, (out, t2, t3), num_threads=5)
12966 self._run_py_multithread_fn(
[all …]
|